__init__(self, model, input_record, prediction_layer, output_dims, subtract_log_odd=True, name='sampling_train', kwargs) (defined in caffe2.python.layers.sampling_train.SamplingTrain) | caffe2.python.layers.sampling_train.SamplingTrain | |
caffe2::python::layers::layers::ModelLayer.__init__(self, model, prefix, input_record, predict_input_record_fields=None, tags=None, kwargs) | caffe2.python.layers.layers.ModelLayer | |
add_eval_ops(self, net) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
add_init_params(self, init_net) | caffe2.python.layers.layers.ModelLayer | |
add_operators(self, net, init_net=None, context=InstantiationContext.TRAINING) | caffe2.python.layers.layers.ModelLayer | |
add_ops(self, net) (defined in caffe2.python.layers.sampling_train.SamplingTrain) | caffe2.python.layers.sampling_train.SamplingTrain | |
add_ops_to_accumulate_pred(self, net) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
add_param_copy_operators(self, net) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
add_train_ops(self, net) (defined in caffe2.python.layers.sampling_train.SamplingTrain) | caffe2.python.layers.sampling_train.SamplingTrain | |
create_param(self, param_name, shape, initializer, optimizer, ps_param=None, regularizer=None) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
eval_output_schema (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
export_output_for_metrics(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
export_params_for_metrics(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
get_fp16_compatible_parameters(self) | caffe2.python.layers.layers.ModelLayer | |
get_memory_usage(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
get_next_blob_reference(self, name) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
get_parameters(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
get_type(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
input_record(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
kwargs (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
model (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
name (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
output_schema (defined in caffe2.python.layers.sampling_train.SamplingTrain) | caffe2.python.layers.sampling_train.SamplingTrain | |
output_schema(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
output_schema(self, output_schema) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
params (defined in caffe2.python.layers.sampling_train.SamplingTrain) | caffe2.python.layers.sampling_train.SamplingTrain | |
precomputation_object_only (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
precomputation_request_only (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
predict_input_record(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
predict_output_schema(self) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
predict_output_schema(self, output_schema) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
request_only (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
subtract_log_odd (defined in caffe2.python.layers.sampling_train.SamplingTrain) | caffe2.python.layers.sampling_train.SamplingTrain | |
tags (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |