__init__(self, model, input_record, name='layer_normalization', scale_optim=None, bias_optim=None, epsilon=1e-4, axis=1, use_layer_norm_op=True, kwargs) (defined in caffe2.python.layers.layer_normalization.LayerNormalization) | caffe2.python.layers.layer_normalization.LayerNormalization | |
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.layer_normalization.LayerNormalization) | caffe2.python.layers.layer_normalization.LayerNormalization | |
add_ops_to_accumulate_pred(self, net) (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
add_ops_with_layer_norm_op(self, net) (defined in caffe2.python.layers.layer_normalization.LayerNormalization) | caffe2.python.layers.layer_normalization.LayerNormalization | |
add_ops_without_layer_norm_op(self, net) (defined in caffe2.python.layers.layer_normalization.LayerNormalization) | caffe2.python.layers.layer_normalization.LayerNormalization | |
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 | |
axis (defined in caffe2.python.layers.layer_normalization.LayerNormalization) | caffe2.python.layers.layer_normalization.LayerNormalization | |
bias (defined in caffe2.python.layers.layer_normalization.LayerNormalization) | caffe2.python.layers.layer_normalization.LayerNormalization | |
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 | |
epsilon (defined in caffe2.python.layers.layer_normalization.LayerNormalization) | caffe2.python.layers.layer_normalization.LayerNormalization | |
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 | |
input_shape (defined in caffe2.python.layers.layer_normalization.LayerNormalization) | caffe2.python.layers.layer_normalization.LayerNormalization | |
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.layer_normalization.LayerNormalization) | caffe2.python.layers.layer_normalization.LayerNormalization | |
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 | |
scale (defined in caffe2.python.layers.layer_normalization.LayerNormalization) | caffe2.python.layers.layer_normalization.LayerNormalization | |
tags (defined in caffe2.python.layers.layers.ModelLayer) | caffe2.python.layers.layers.ModelLayer | |
use_layer_norm_op (defined in caffe2.python.layers.layer_normalization.LayerNormalization) | caffe2.python.layers.layer_normalization.LayerNormalization | |