Home
last modified time | relevance | path

Searched defs:exponential_avg_factor (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/tests/
Dfused_batchnorm_test.py48 exponential_avg_factor, data_format): argument
133 exponential_avg_factor): argument
210 def testLearning(self, data_format, exponential_avg_factor): argument
215 exponential_avg_factor): argument
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op_test.cc79 static void VerifyTensorsClose(const float exponential_avg_factor, in VerifyTensorsClose()
224 void VerifyFusedBatchNorm(const float exponential_avg_factor, in VerifyFusedBatchNorm()
231 Tensor* batch_mean, Tensor* batch_var) { in VerifyFusedBatchNorm()
277 Tensor* batch_mean, Tensor* batch_var) { in VerifyFusedBatchNorm()
477 const float exponential_avg_factor = 1.0; in TYPED_TEST_P() local
483 const float exponential_avg_factor = 0.5; in TYPED_TEST_P() local
489 const float exponential_avg_factor = 1.0; in TYPED_TEST_P() local
495 const float exponential_avg_factor = 0.5; in TYPED_TEST_P() local
Dmkl_fused_batch_norm_op.cc646 float exponential_avg_factor; in MklFusedBatchNormOp() local
/external/tensorflow/tensorflow/python/ops/
Dnn_fused_batchnorm_test.py66 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
Dnn_impl.py1594 exponential_avg_factor=1.0): argument
/external/tensorflow/tensorflow/core/ops/
Dnn_ops_test.cc185 string data_format) { in TEST()
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op.cc1190 float exponential_avg_factor; in FusedBatchNormOpBase() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc1146 float exponential_avg_factor; in FusedBatchNormShape() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2011 float exponential_avg_factor = in matchAndRewrite() local