Home
last modified time | relevance | path

Searched refs:LayerNormalization (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization_test.py539 keras.layers.LayerNormalization,
546 keras.layers.LayerNormalization,
553 keras.layers.LayerNormalization,
558 keras.layers.LayerNormalization,
565 keras.layers.LayerNormalization,
569 keras.layers.LayerNormalization,
573 keras.layers.LayerNormalization,
579 layer = keras.layers.LayerNormalization(scale=False, center=False)
584 layer = keras.layers.LayerNormalization()
591 layer = keras.layers.LayerNormalization(
[all …]
Dnormalization.py976 class LayerNormalization(Layer): class
1110 super(LayerNormalization, self).__init__(**kwargs)
1310 base_config = super(LayerNormalization, self).get_config()
D__init__.py172 from tensorflow.python.keras.layers.normalization import LayerNormalization
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dlayer_correctness_test.py126 ('LayerNormalization', normalization.LayerNormalization, (2, 2)),
128 lambda: normalization.LayerNormalization(axis=1), (2, 2, 2)),
/external/tensorflow/tensorflow/python/keras/benchmarks/keras_examples_benchmarks/
Dtext_classification_transformer_benchmark_test.py204 self.layernorm1 = tf.keras.layers.LayerNormalization(epsilon=1e-6)
205 self.layernorm2 = tf.keras.layers.LayerNormalization(epsilon=1e-6)
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.keras.layers.-layer-normalization.pbtxt1 path: "tensorflow.keras.layers.LayerNormalization"
3 is_instance: "<class \'tensorflow.python.keras.layers.normalization.LayerNormalization\'>"
Dtensorflow.keras.layers.pbtxt256 name: "LayerNormalization"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.keras.layers.-layer-normalization.pbtxt1 path: "tensorflow.keras.layers.LayerNormalization"
3 is_instance: "<class \'tensorflow.python.keras.layers.normalization.LayerNormalization\'>"
Dtensorflow.keras.layers.pbtxt264 name: "LayerNormalization"
/external/tensorflow/tensorflow/python/keras/benchmarks/
Deager_microbenchmarks_test.py192 tf.keras.layers.LayerNormalization(), (1, 1),
/external/tensorflow/tensorflow/python/keras/benchmarks/layer_benchmarks/
Dlayer_benchmarks_test.py140 ("LayerNormalization_small_shape", tf.keras.layers.LayerNormalization,
/external/tensorflow/
DRELEASE.md2025 * Standardize the LayerNormalization API by replacing the args `norm_axis`
2358 * Standardize the LayerNormalization API by replacing the args `norm_axis`