Searched refs:variance_val (Results 1 – 3 of 3) sorted by relevance
121 self.variance_val = variance171 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)
111 self.variance_val = variance161 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])
1289 Value mean_val, variance_val; in matchAndRewrite() local1298 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()