 |
Caffe2 - Python API
A deep learning, cross platform ML framework
|
|
Caffe2 - Python API
A deep learning, cross platform ML framework
|
Public Member Functions | |
| def | __init__ (self, model, input_record, dims, name='expand_dims', kwargs) |
| def | add_ops (self, net) |
 Public Member Functions inherited from layers.layers.ModelLayer | |
| def | __init__ (self, model, prefix, input_record, predict_input_record_fields=None, tags=None, kwargs) |
| def | get_type (self) |
| def | predict_input_record (self) |
| def | input_record (self) |
| def | predict_output_schema (self) |
| def | predict_output_schema (self, output_schema) |
| def | output_schema (self) |
| def | output_schema (self, output_schema) |
| def | get_parameters (self) |
| def | get_fp16_compatible_parameters (self) |
| def | get_memory_usage (self) |
| def | add_operators (self, net, init_net=None, context=InstantiationContext.TRAINING) |
| def | add_ops (self, net) |
| def | add_eval_ops (self, net) |
| def | add_train_ops (self, net) |
| def | add_calibration_ops (self, net) |
Public Attributes | |
| dims | |
| output_schema | |
| Public Attributes inherited from layers.layers.ModelLayer | |
| name | |
| model | |
| kwargs | |
| request_only | |
| eval_output_schema | |
| tags | |
| params | |
Definition at line 14 of file expand_dims.py.
1.8.14