Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dbatch_norm_op.cc39 float variance_epsilon; in BatchNormOp() local
41 context->GetAttr("variance_epsilon", &variance_epsilon)); in BatchNormOp()
42 variance_epsilon_ = T(variance_epsilon); in BatchNormOp()
89 float variance_epsilon; in BatchNormGradOp() local
91 context->GetAttr("variance_epsilon", &variance_epsilon)); in BatchNormGradOp()
92 variance_epsilon_ = T(variance_epsilon); in BatchNormGradOp()
187 T variance_epsilon, bool scale_after_normalization, \
243 typename TTypes<T, 4>::ConstTensor out_backprop, T variance_epsilon, \
Dbatch_norm_op.h32 typename TTypes<T>::ConstVec gamma, T variance_epsilon, in operator()
55 ((var + var.constant(variance_epsilon)).rsqrt() * gamma) in operator()
64 ((var + var.constant(variance_epsilon)).rsqrt()) in operator()
80 T variance_epsilon, bool scale_after_normalization, in operator()
117 scratch1.device(d) = (var + var.constant(variance_epsilon)).rsqrt(); in operator()
143 (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/python/ops/
Dnn_impl.py1136 variance_epsilon, argument
1181 inv = math_ops.rsqrt(variance + variance_epsilon)
1267 variance_epsilon=None, argument
1305 else None, variance_epsilon, name)
1315 variance_epsilon, argument
1348 variance_epsilon=variance_epsilon,
/external/tensorflow/tensorflow/contrib/distributions/python/ops/bijectors/
Dbatch_normalization.py51 variance_epsilon, argument
77 rescale = math_ops.sqrt(variance + variance_epsilon)
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_old_batch_norms.cc75 const float variance_epsilon = batch_norm_node.attr().at(epsilon_attr).f(); in GetScaleAndOffsetValues() local
90 (1.0f / sqrtf(variance.flat<float>()(i) + variance_epsilon)) * in GetScaleAndOffsetValues()
96 (1.0f / sqrtf(variance.flat<float>()(i) + variance_epsilon)); in GetScaleAndOffsetValues()
/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/grappler/optimizers/
Dremapper.cc711 NodeDef* variance_epsilon = optimized_graph->add_node(); in AddBatchNormNodes() local
714 variance_epsilon)); in AddBatchNormNodes()
715 variance_epsilon->set_device(fused_node.device()); in AddBatchNormNodes()
724 *variance_plus_epsilon->add_input() = variance_epsilon->name(); in AddBatchNormNodes()
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.nn.pbtxt49 …argspec: "args=[\'input\', \'mean\', \'variance\', \'beta\', \'gamma\', \'variance_epsilon\', \'sc…
53 …argspec: "args=[\'x\', \'mean\', \'variance\', \'offset\', \'scale\', \'variance_epsilon\', \'name…
Dtensorflow.raw_ops.pbtxt357 …argspec: "args=[\'t\', \'m\', \'v\', \'beta\', \'gamma\', \'variance_epsilon\', \'scale_after_norm…
361 …argspec: "args=[\'t\', \'m\', \'v\', \'gamma\', \'backprop\', \'variance_epsilon\', \'scale_after_…
2421 …eta_max\', \'gamma\', \'gamma_min\', \'gamma_max\', \'out_type\', \'variance_epsilon\', \'scale_af…
2477 …_max\', \'output_range_given\', \'given_y_min\', \'given_y_max\', \'variance_epsilon\', \'min_sepa…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt45 …argspec: "args=[\'t\', \'m\', \'v\', \'beta\', \'gamma\', \'variance_epsilon\', \'scale_after_norm…
49 …argspec: "args=[\'x\', \'mean\', \'variance\', \'offset\', \'scale\', \'variance_epsilon\', \'name…
Dtensorflow.raw_ops.pbtxt357 …argspec: "args=[\'t\', \'m\', \'v\', \'beta\', \'gamma\', \'variance_epsilon\', \'scale_after_norm…
361 …argspec: "args=[\'t\', \'m\', \'v\', \'gamma\', \'backprop\', \'variance_epsilon\', \'scale_after_…
2421 …eta_max\', \'gamma\', \'gamma_min\', \'gamma_max\', \'out_type\', \'variance_epsilon\', \'scale_af…
2477 …_max\', \'output_range_given\', \'given_y_min\', \'given_y_max\', \'variance_epsilon\', \'min_sepa…
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers.py2317 variance_epsilon = 1e-12 if dtype != dtypes.float16 else 1e-3
2324 variance_epsilon=variance_epsilon)
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization.py997 variance_epsilon=self.epsilon)
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v0.pbtxt5490 name: "variance_epsilon"
5545 name: "variance_epsilon"
5601 name: "variance_epsilon"
5658 name: "variance_epsilon"
5715 name: "variance_epsilon"
5788 name: "variance_epsilon"
5859 name: "variance_epsilon"
5931 name: "variance_epsilon"
6004 name: "variance_epsilon"
6077 name: "variance_epsilon"
[all …]
Dops_history.v2.pbtxt9727 name: "variance_epsilon"
9789 name: "variance_epsilon"
9852 name: "variance_epsilon"
9915 name: "variance_epsilon"
9991 name: "variance_epsilon"
10069 name: "variance_epsilon"
10148 name: "variance_epsilon"
10227 name: "variance_epsilon"
46414 name: "variance_epsilon"
46523 name: "variance_epsilon"
[all …]
Dops_history.v1.pbtxt9727 name: "variance_epsilon"
9789 name: "variance_epsilon"
9852 name: "variance_epsilon"
9915 name: "variance_epsilon"
9991 name: "variance_epsilon"
10069 name: "variance_epsilon"
10148 name: "variance_epsilon"
10227 name: "variance_epsilon"
46997 name: "variance_epsilon"
47106 name: "variance_epsilon"
[all …]
/external/tensorflow/tensorflow/core/graph/
Dgraph_constructor_test.cc1124 key: "variance_epsilon"
/external/tensorflow/tensorflow/go/op/
Dwrappers.go7865 …ut, gamma_min tf.Output, gamma_max tf.Output, out_type tf.DataType, variance_epsilon float32, scal…
7869 …attrs := map[string]interface{}{"out_type": out_type, "variance_epsilon": variance_epsilon, "scale…
20018 ….Output, m tf.Output, v tf.Output, beta tf.Output, gamma tf.Output, variance_epsilon float32, scal…
20022 …attrs := map[string]interface{}{"variance_epsilon": variance_epsilon, "scale_after_normalization":…
38937 …put, m tf.Output, v tf.Output, gamma tf.Output, backprop tf.Output, variance_epsilon float32, scal…
38941 …attrs := map[string]interface{}{"variance_epsilon": variance_epsilon, "scale_after_normalization":…
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt3669 name: "variance_epsilon"
3749 name: "variance_epsilon"
23894 name: "variance_epsilon"
25512 name: "variance_epsilon"