Caffe2 - Python API
A deep learning, cross platform ML framework
Classes | Functions
SparseTransformer Namespace Reference

Module caffe2.experiments.python.SparseTransformer. More...

Classes

class  NetDefNode
 

Functions

def maskNallocate (weight_name)
 
def transFCRelu (cur, id2node, name2id, ops, model)
 
def Prune2Sparse (cur, id2node, name2id, ops, model)
 
def net2list (net_root)
 
def netbuilder (model)
 

Detailed Description

Module caffe2.experiments.python.SparseTransformer.

Function Documentation

def SparseTransformer.maskNallocate (   weight_name)
Combine mask and weights
create wcsr, iw, jw, return their names

Definition at line 57 of file SparseTransformer.py.

def SparseTransformer.net2list (   net_root)
Use topological order(BFS) to print the op of a net in a list

Definition at line 156 of file SparseTransformer.py.

def SparseTransformer.transFCRelu (   cur,
  id2node,
  name2id,
  ops,
  model 
)
Add trans before and after this FC_Prune->(Relu)->FC_Prune chain.

Definition at line 73 of file SparseTransformer.py.