Searched refs:exponential_avg_factor_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 41 ctx, ctx->GetAttr("exponential_avg_factor", &exponential_avg_factor_)); in FusedBatchNormOp()
134 if (exponential_avg_factor_ == 1.0f) { in CompileImpl()
140 xla::ScalarLike(old_mean, 1.0f - exponential_avg_factor_); in CompileImpl()
141 xla::XlaOp beta = xla::ScalarLike(old_mean, exponential_avg_factor_); in CompileImpl()
199 float exponential_avg_factor_; member in tensorflow::__anonfc91c4b10111::FusedBatchNormOp
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 649 exponential_avg_factor_ = static_cast<U>(exponential_avg_factor); in MklFusedBatchNormOp()
886 if (exponential_avg_factor_ == U(1.0)) { in Compute()
893 U one_minus_factor = U(1.0) - exponential_avg_factor_; in Compute()
896 exponential_avg_factor_ * mean_data[k]; in Compute()
898 exponential_avg_factor_ * in Compute()
919 U exponential_avg_factor_; member in tensorflow::MklFusedBatchNormOp
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 1193 exponential_avg_factor_ = U(exponential_avg_factor); in FusedBatchNormOpBase()
1320 side_input, epsilon_, exponential_avg_factor_, activation_mode_, y, in ComputeWithReservedSpace()
1326 side_input, epsilon_, exponential_avg_factor_, activation_mode_, y, in ComputeWithReservedSpace()
1338 U exponential_avg_factor_; member in tensorflow::FusedBatchNormOpBase
|