Home
last modified time | relevance | path

Searched refs:variance_val (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/python/keras/layers/preprocessing/
Dnormalization.py121 self.variance_val = variance
171 if (self.mean_val is not None and self.variance_val is not None):
173 variance_val = self.variance_val * np.ones(mean_and_var_shape)
175 self.variance.assign(variance_val)
Dnormalization_v1.py111 self.variance_val = variance
161 if (self.mean_val is not None and self.variance_val is not None):
163 variance_val = self.variance_val * np.ones(mean_and_var_shape)
164 self.set_weights([mean_val, variance_val])
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dlegalize_tf.cc1289 Value mean_val, variance_val; in matchAndRewrite() local
1298 variance_val = getTosaConstTensorSingleF32(rewriter, tf_batchnorm_op, 1.0); in matchAndRewrite()
1300 variance_val = tf_batchnorm_op.variance(); in matchAndRewrite()
1315 rewriter.create<tosa::AddOp>(op->getLoc(), variance_val.getType(), in matchAndRewrite()
1316 variance_val, epsilon_const.getResult()); in matchAndRewrite()
1319 op->getLoc(), variance_val.getType(), op2_add_var_epsilon.getResult()); in matchAndRewrite()