Searched refs:variance_type (Results 1 – 2 of 2) sorted by relevance
117 auto variance_type = in matchAndRewrite() local119 if (!input_type || !variance_type) { in matchAndRewrite()122 auto fp_type = variance_type.getElementType().dyn_cast<FloatType>(); in matchAndRewrite()132 bn_op.variance(), variance_type, rewriter); in matchAndRewrite()
1285 RankedTensorType variance_type = in matchAndRewrite() local1287 if (!variance_type || !mean_type) return failure(); in matchAndRewrite()1297 if (variance_type.getNumElements() == 0) { in matchAndRewrite()1304 RankedTensorType::get({1}, variance_type.getElementType()); in matchAndRewrite()1359 RankedTensorType variance_type = in matchAndRewrite() local1361 if (!variance_type) return failure(); in matchAndRewrite()1364 RankedTensorType::get({1}, variance_type.getElementType()); in matchAndRewrite()