Searched defs:exponential_avg_factor (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_fused_batch_norm_op_test.cc | 80 static void VerifyTensorsClose(const float exponential_avg_factor, in VerifyTensorsClose()
225 void VerifyFusedBatchNorm(const float exponential_avg_factor, in VerifyFusedBatchNorm()
232 Tensor* batch_mean, Tensor* batch_var) { in VerifyFusedBatchNorm()
278 Tensor* batch_mean, Tensor* batch_var) { in VerifyFusedBatchNorm()
481 const float exponential_avg_factor = 1.0; in TYPED_TEST_P() local
487 const float exponential_avg_factor = 0.5; in TYPED_TEST_P() local
493 const float exponential_avg_factor = 1.0; in TYPED_TEST_P() local
499 const float exponential_avg_factor = 0.5; in TYPED_TEST_P() local
|
| H A D | mkl_fused_batch_norm_op.cc | 670 float exponential_avg_factor; in MklFusedBatchNormOp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | fused_batchnorm_test.py | 45 exponential_avg_factor, data_format): argument
130 exponential_avg_factor): argument
207 def testLearning(self, data_format, exponential_avg_factor): argument
212 exponential_avg_factor): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_fused_batchnorm_test.py | 66 exponential_avg_factor=1.0, argument
108 exponential_avg_factor, epsilon, data_format): argument
147 exponential_avg_factor=1.0, argument
228 exponential_avg_factor=1.0, argument
291 exponential_avg_factor=1.0, argument
|
| H A D | nn_impl.py | 1609 exponential_avg_factor=1.0): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | nn_ops_test.cc | 185 string data_format) { in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | fused_batch_norm_op.cc | 1216 float exponential_avg_factor; in FusedBatchNormOpBase() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | common_shape_fns.cc | 1201 float exponential_avg_factor; in FusedBatchNormShape() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| H A D | legalize_tf.cc | 2410 float exponential_avg_factor = in matchAndRewrite() local
|