| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| H A D | element_wise_row_vectorization.hlo | 56 …etadata={op_type="FusedBatchNormGradV3" op_name="gradient_tape/foo/batch_normalization/FusedBatchN…
57 …etadata={op_type="FusedBatchNormGradV3" op_name="gradient_tape/foo/batch_normalization/FusedBatchN…
61 …%broadcast.185), metadata={op_type="FusedBatchNormV3" op_name="foo/batch_normalization/FusedBatchN…
63 …%broadcast.185), metadata={op_type="FusedBatchNormV3" op_name="foo/batch_normalization/FusedBatchN…
64 … %multiply.154), metadata={op_type="FusedBatchNormV3" op_name="foo/batch_normalization/FusedBatchN…
65 … %multiply.153), metadata={op_type="FusedBatchNormV3" op_name="foo/batch_normalization/FusedBatchN…
66 …constant(0.001), metadata={op_type="FusedBatchNormV3" op_name="foo/batch_normalization/FusedBatchN…
68 …%broadcast.184), metadata={op_type="FusedBatchNormV3" op_name="foo/batch_normalization/FusedBatchN…
69 …72]{0} %add.14), metadata={op_type="FusedBatchNormV3" op_name="foo/batch_normalization/FusedBatchN…
70 …etadata={op_type="FusedBatchNormGradV3" op_name="gradient_tape/foo/batch_normalization/FusedBatchN…
[all …]
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/batchnormalization/impl/NEON/ |
| H A D | fp32.cpp | 43 void batch_normalization(ITensor *src, ITensor *dst, const ITensor *mean, const ITensor *var, const… in batch_normalization() function
128 { ActivationLayerInfo::ActivationFunction::RELU, &batch_normalization<detail::relu<float, 4>> },
129 …{ ActivationLayerInfo::ActivationFunction::BOUNDED_RELU, &batch_normalization<detail::brelu<float,…
130 …{ ActivationLayerInfo::ActivationFunction::LU_BOUNDED_RELU, &batch_normalization<detail::lubrelu<f…
144 …batch_normalization<detail::dummy<float, 4>>(src, dst, mean, var, beta, gamma, epsilon, act_info, … in fp32_neon_batch_normalization()
|
| H A D | fp16.cpp | 44 void batch_normalization(ITensor *src, ITensor *dst, const ITensor *mean, const ITensor *var, const… in batch_normalization() function
129 …{ ActivationLayerInfo::ActivationFunction::RELU, &batch_normalization<detail::relu<float16_t, 8>> …
130 …{ ActivationLayerInfo::ActivationFunction::BOUNDED_RELU, &batch_normalization<detail::brelu<float1…
131 …{ ActivationLayerInfo::ActivationFunction::LU_BOUNDED_RELU, &batch_normalization<detail::lubrelu<f…
145 …batch_normalization<detail::dummy<float16_t, 8>>(src, dst, mean, var, beta, gamma, epsilon, act_in… in fp16_neon_batch_normalization()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/normalization/ |
| H A D | BUILD | 32 ":batch_normalization",
39 name = "batch_normalization",
40 srcs = ["batch_normalization.py"],
73 ":batch_normalization",
107 ":batch_normalization",
|
| H A D | layer_normalization.py | 289 outputs = nn.batch_normalization(
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/layers/ |
| H A D | normalization.py | 32 return normalization.batch_normalization
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/ |
| H A D | api_def_BatchNormWithGlobalNormalization.pbtxt | 55 This op is deprecated. Prefer `tf.nn.batch_normalization`.
|
| H A D | api_def_BatchNormWithGlobalNormalizationGrad.pbtxt | 84 This op is deprecated. See `tf.nn.batch_normalization`.
|
| H A D | api_def_QuantizedBatchNormWithGlobalNormalization.pbtxt | 116 `tf.nn.batch_normalization`.
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | batch_norm_benchmark.py | 53 return nn_impl.batch_normalization(tensor, mean, variance, beta, gamma if
|
| H A D | nn_impl.py | 1528 def batch_normalization(x, function
1755 return batch_normalization(t, m, v, beta, gamma if scale_after_normalization
|
| H A D | nn_batchnorm_test.py | 64 return nn_impl.batch_normalization(x, m, v, beta if
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.layers.-batch-normalization.pbtxt | 5 is_instance: "<class \'keras.layers.normalization.batch_normalization.BatchNormalizationBase\'>"
|
| H A D | tensorflow.layers.pbtxt | 96 name: "batch_normalization"
|
| H A D | tensorflow.nn.pbtxt | 56 name: "batch_normalization"
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/ |
| H A D | backend.py | 2891 normed = nn.batch_normalization(x, mean, var, beta, gamma, epsilon)
2933 normed = nn.batch_normalization(x, broadcast_mean, broadcast_var,
3008 def batch_normalization(x, mean, var, beta, gamma, axis=-1, epsilon=1e-3): function
3063 return nn.batch_normalization(x, mean, var, beta, gamma, epsilon)
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.nn.pbtxt | 60 name: "batch_normalization"
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/grappler/ |
| H A D | layout_optimizer_test.py | 1311 y = nn.batch_normalization(
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | ops.pbtxt | 4599 explanation: "Use tf.nn.batch_normalization()"
4679 explanation: "Use tf.nn.batch_normalization()"
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| H A D | tf_generated_ops.td | 1247 This op is deprecated. Prefer `tf.nn.batch_normalization`.
|