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 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.