Home
last modified time | relevance | path

Searched refs:variance_epsilon (Results 1 – 25 of 38) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Dbatch_norm_op.cc36 float variance_epsilon; in BatchNormOp() local
38 context->GetAttr("variance_epsilon", &variance_epsilon)); in BatchNormOp()
39 variance_epsilon_ = T(variance_epsilon); in BatchNormOp()
86 float variance_epsilon; in BatchNormGradOp() local
88 context->GetAttr("variance_epsilon", &variance_epsilon)); in BatchNormGradOp()
89 variance_epsilon_ = T(variance_epsilon); in BatchNormGradOp()
184 T variance_epsilon, bool scale_after_normalization, \
229 typename TTypes<T, 4>::ConstTensor out_backprop, T variance_epsilon, \
Dbatch_norm_op.h32 typename TTypes<T>::ConstVec gamma, T variance_epsilon, in operator()
49 ((var + var.constant(variance_epsilon)).rsqrt() * gamma) in operator()
58 ((var + var.constant(variance_epsilon)).rsqrt()) in operator()
74 T variance_epsilon, bool scale_after_normalization, in operator()
104 scratch1.device(d) = (var + var.constant(variance_epsilon)).rsqrt(); in operator()
130 (var + var.constant(variance_epsilon)); in operator()
Dquantized_batch_norm_op.cc37 float variance_epsilon, bool scale_after_normalization, in ReferenceBatchNorm() argument
69 sqrtf(var_value + variance_epsilon)) * in ReferenceBatchNorm()
74 sqrtf(var_value + variance_epsilon)) + in ReferenceBatchNorm()
100 float variance_epsilon, bool scale_after_normalization, in FixedPointBatchNorm() argument
133 scale_value = (1.0f / sqrtf(var_value + variance_epsilon)) * gamma_value; in FixedPointBatchNorm()
135 scale_value = (1.0f / sqrtf(var_value + variance_epsilon)); in FixedPointBatchNorm()
Dquantized_instance_norm.cc140 float variance_epsilon, float* minimum, float* maximum) { in MinAndMax() argument
143 const float32x4_t eps = vdupq_n_f32(variance_epsilon); in MinAndMax()
196 const float* variance_ptr, float variance_epsilon, in InstanceNorm() argument
198 const float32x4_t eps = vdupq_n_f32(variance_epsilon); in InstanceNorm()
/external/tensorflow/tensorflow/security/advisory/
Dtfsa-2022-102.md14 variance_epsilon = 1e-05
19 …ge_given, given_y_min=given_y_min, given_y_max=given_y_max, variance_epsilon=variance_epsilon, min…
Dtfsa-2021-036.md34 variance_epsilon=0.1, scale_after_normalization=True)
Dtfsa-2021-035.md34 variance_epsilon=0.1, scale_after_normalization=True)
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
DBatchNormWithGlobalNormalization.pbtxt50 name: "variance_epsilon"
112 name: "variance_epsilon"
175 name: "variance_epsilon"
238 name: "variance_epsilon"
DBatchNormWithGlobalNormalizationGrad.pbtxt66 name: "variance_epsilon"
144 name: "variance_epsilon"
223 name: "variance_epsilon"
302 name: "variance_epsilon"
DQuantizedInstanceNorm.pbtxt62 name: "variance_epsilon"
137 name: "variance_epsilon"
DQuantizedBatchNormWithGlobalNormalization.pbtxt102 name: "variance_epsilon"
211 name: "variance_epsilon"
/external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/
DBatchNormWithGlobalNormalization.pbtxt50 name: "variance_epsilon"
112 name: "variance_epsilon"
175 name: "variance_epsilon"
238 name: "variance_epsilon"
DBatchNormWithGlobalNormalizationGrad.pbtxt66 name: "variance_epsilon"
144 name: "variance_epsilon"
223 name: "variance_epsilon"
302 name: "variance_epsilon"
DQuantizedInstanceNorm.pbtxt62 name: "variance_epsilon"
137 name: "variance_epsilon"
DQuantizedBatchNormWithGlobalNormalization.pbtxt102 name: "variance_epsilon"
211 name: "variance_epsilon"
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_patterns.td38 …$t, $m, $v, $beta, $gamma, F32Attr:$variance_epsilon, ConstBoolAttrFalse:$scale_after_normalizatio…
40 (TF_MulOp $t, (TF_RsqrtOp:$rsqrt (TF_AddOp $v, (TF_ConstOp $variance_epsilon)))),
44 …$t, $m, $v, $beta, $gamma, F32Attr:$variance_epsilon, ConstBoolAttrTrue:$scale_after_normalization…
46 … (TF_MulOp $t, (TF_MulOp:$mul (TF_RsqrtOp (TF_AddOp $v, (TF_ConstOp $variance_epsilon))), $gamma)),
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_old_batch_norms.cc74 const float variance_epsilon = batch_norm_node.attr().at(epsilon_attr).f(); in GetScaleAndOffsetValues() local
89 (1.0f / sqrtf(variance.flat<float>()(i) + variance_epsilon)) * in GetScaleAndOffsetValues()
95 (1.0f / sqrtf(variance.flat<float>()(i) + variance_epsilon)); in GetScaleAndOffsetValues()
/external/tensorflow/tensorflow/python/ops/
Dnn_impl.py1533 variance_epsilon, argument
1588 inv = math_ops.rsqrt(variance + variance_epsilon)
1712 variance_epsilon=None, argument
1756 else None, variance_epsilon, name)
1767 variance_epsilon, argument
1805 variance_epsilon=variance_epsilon,
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_QuantizedInstanceNorm.pbtxt60 name: "variance_epsilon"
Dapi_def_BatchNormWithGlobalNormalization.pbtxt41 name: "variance_epsilon"
Dapi_def_BatchNormWithGlobalNormalizationGrad.pbtxt70 name: "variance_epsilon"
Dapi_def_QuantizedBatchNormWithGlobalNormalization.pbtxt101 name: "variance_epsilon"
/external/tensorflow/tensorflow/core/ir/importexport/tests/roundtrip/
Dgraph-version-info.pbtxt156 key: "variance_epsilon"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dgraph-version-info.pbtxt159 key: "variance_epsilon"
/external/tensorflow/tensorflow/python/keras/layers/normalization/
Dlayer_normalization.py295 variance_epsilon=self.epsilon)

12