Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | Public Attributes | Static Public Attributes | List of all members
caffe2.python.layers.tags.Tags Class Reference
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)
 

Public Attributes

 tags
 

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 GRADIENT_FROM_PS = 'gradient_from_ps'
 
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:'
 
tuple DT_TAGS = (SPARSE_SHARDED, SPARSE_DONT_SHARD, COMPONENT)
 
string PREDICTION_SCHEMA = 'prediction_schema'
 

Detailed Description

Definition at line 46 of file tags.py.


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