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.py49 exponential_avg_factor, data_format): argument
134 exponential_avg_factor): argument
211 def testLearning(self, data_format, exponential_avg_factor): argument
216 exponential_avg_factor): argument
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op_test.cc80 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
Dmkl_fused_batch_norm_op.cc653 float exponential_avg_factor; in MklFusedBatchNormOp() local
/external/tensorflow/tensorflow/python/ops/
Dnn_fused_batchnorm_test.py67 exponential_avg_factor=1.0, argument
109 exponential_avg_factor, epsilon, data_format): argument
148 exponential_avg_factor=1.0, argument
229 exponential_avg_factor=1.0, argument
292 exponential_avg_factor=1.0, argument
Dnn_impl.py1587 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.cc1211 float exponential_avg_factor; in FusedBatchNormOpBase() local
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc1186 float exponential_avg_factor; in FusedBatchNormShape() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf.cc2624 float exponential_avg_factor = in matchAndRewrite() local