__init__(self, model, input_record, num_to_collect, name='reservoir_sampling', kwargs) (defined in caffe2.python.layers.reservoir_sampling.ReservoirSampling) | caffe2.python.layers.reservoir_sampling.ReservoirSampling | |
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.reservoir_sampling.ReservoirSampling) | caffe2.python.layers.reservoir_sampling.ReservoirSampling | |
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.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
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 | |
extra_input_blobs (defined in caffe2.python.layers.reservoir_sampling.ReservoirSampling) | caffe2.python.layers.reservoir_sampling.ReservoirSampling | |
extra_output_blobs (defined in caffe2.python.layers.reservoir_sampling.ReservoirSampling) | caffe2.python.layers.reservoir_sampling.ReservoirSampling | |
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 | |
mutex (defined in caffe2.python.layers.reservoir_sampling.ReservoirSampling) | caffe2.python.layers.reservoir_sampling.ReservoirSampling | |
name (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
num_to_collect (defined in caffe2.python.layers.reservoir_sampling.ReservoirSampling) | caffe2.python.layers.reservoir_sampling.ReservoirSampling | |
num_visited_blob (defined in caffe2.python.layers.reservoir_sampling.ReservoirSampling) | caffe2.python.layers.reservoir_sampling.ReservoirSampling | |
output_schema (defined in caffe2.python.layers.reservoir_sampling.ReservoirSampling) | caffe2.python.layers.reservoir_sampling.ReservoirSampling | |
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.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
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 | |
reservoir (defined in caffe2.python.layers.reservoir_sampling.ReservoirSampling) | caffe2.python.layers.reservoir_sampling.ReservoirSampling | |
tags (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |