Searched refs:variance_epsilon_ (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | batch_norm_op.cc | 39 variance_epsilon_ = T(variance_epsilon); in BatchNormOp()
73 var.vec<T>(), beta.vec<T>(), gamma.vec<T>(), variance_epsilon_, in Compute()
78 T variance_epsilon_; member in tensorflow::BatchNormOp
89 variance_epsilon_ = T(variance_epsilon); in BatchNormGradOp()
154 variance_epsilon_, scale_after_normalization_, dx->tensor<T, 4>(), in Compute()
160 T variance_epsilon_; member in tensorflow::BatchNormGradOp
|
| D | quantized_instance_norm.cc | 259 context->GetAttr("variance_epsilon", &variance_epsilon_)); in QuantizedInstanceNorm()
333 variance_epsilon_, &minimum, &maximum); in Compute()
342 variance_epsilon_, minimum, maximum, in Compute()
364 (float_variance + variance_epsilon_) in Compute()
397 float variance_epsilon_; member in tensorflow::QuantizedInstanceNorm
|
| D | quantized_batch_norm_op.cc | 169 context->GetAttr("variance_epsilon", &variance_epsilon_)); in QuantizedBatchNormOp()
215 variance_epsilon_, scale_after_normalization_, in Compute()
230 float variance_epsilon_; member in tensorflow::QuantizedBatchNormOp
|