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) |
Module caffe2.experiments.python.SparseTransformer.
def SparseTransformer.maskNallocate | ( | weight_name | ) |
Combine mask and weights create wcsr, iw, jw, return their names
Definition at line 42 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 141 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 58 of file SparseTransformer.py.