Inheritance diagram for caffe2.python.layers.tags.Tags:

Public Member Functions
def	__init__ (self, tags)

def	__enter__ (self)

def	__exit__ (self, type, value, traceback)

def	__call__ (self, func)

Static Public Attributes
string	EXCLUDE_FROM_TRAIN = 'exclude_from_train'

string	EXCLUDE_FROM_EVAL = 'exclude_from_eval'

string	EXCLUDE_FROM_PREDICTION = 'exclude_from_prediction'

string	EXCLUDE_FROM_ACCUMULATE_PRED = 'exclude_from_accumulate_pred'

string	PREPROCESSING = 'preprocessing'

string	HANDLE_AS_SPARSE_LAYER = 'handle_as_sparse_layer'

string	PREFER_GPU = 'prefer_gpu'

string	CPU_ONLY = 'cpu_only'

string	SPARSE_SHARDED = 'sparse_sharded'

string	SPARSE_DONT_SHARD = 'sparse_dont_shard'

string	COMPONENT = 'component:'

string	EXTRA_INFO = 'extra_info:'

string	EMPTY_TAG = 'empty_tag'

tuple	DT_TAGS = (SPARSE_SHARDED, SPARSE_DONT_SHARD, COMPONENT)

string	PREDICTION_SCHEMA = 'prediction_schema'

string	FEATURE_TRANSFORM = 'feature_transform'

string	FEATURE_TRANSFORM_SCHEMA = 'feature_transform_schema'

Detailed Description

Definition at line 31 of file tags.py.

The documentation for this class was generated from the following file:

Public Member Functions