Searched refs:tf_batchnorm_op (Results 1 – 1 of 1) sorted by relevance
1256 auto tf_batchnorm_op = cast<TF::FusedBatchNormOp>(op); in matchAndRewrite() local1259 tf_batchnorm_op.getResult(0).getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()1284 tf_batchnorm_op.mean().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()1286 tf_batchnorm_op.variance().getType().dyn_cast<RankedTensorType>(); in matchAndRewrite()1292 mean_val = getTosaConstTensorSingleF32(rewriter, tf_batchnorm_op, 0); in matchAndRewrite()1294 mean_val = tf_batchnorm_op.mean(); in matchAndRewrite()1298 variance_val = getTosaConstTensorSingleF32(rewriter, tf_batchnorm_op, 1.0); in matchAndRewrite()1300 variance_val = tf_batchnorm_op.variance(); in matchAndRewrite()1306 DenseFPElementsAttr::get(epsilon_type, {tf_batchnorm_op.epsilon()}); in matchAndRewrite()1311 op->getLoc(), tf_batchnorm_op.getResult(0).getType(), tf_batchnorm_op.x(), in matchAndRewrite()[all …]