Lines Matching defs:resultTy

74   ShapedType resultTy = cast<ShapedType>(conv.getType());
77           loc, resultTy, ValueRange({bias, conv}), result, indexingMaps,
78           getNParallelLoopsAttrs(resultTy.getRank()),
95   ShapedType resultTy = cast<ShapedType>(result.getType());
97   const int64_t resultRank = resultTy.getRank();
126   ShapedType resultTy = cast<ShapedType>(result.getType());
127   const int64_t resultRank = resultTy.getRank();
136           loc, resultTy, ValueRange({source}), result, indexingMaps,
137           getNParallelLoopsAttrs(resultTy.getRank()),
185     Location loc, Value input, Value weight, ShapedType resultTy,
193   dynDims.resize(resultTy.getRank());
198     if (resultTy.isDynamicDim(inputDim)) {
215     if (resultTy.isDynamicDim(i) && !dynDims[i])
253     ShapedType resultTy = cast<ShapedType>(op->getResult(0).getType());
256     Type resultETy = resultTy.getElementType();
273     for (int i = 1; i < resultTy.getRank() - 1; i++) {
279         loc, input, weight, resultTy, padAttr.asArrayRef(),
323         for (int i = 1; i < resultTy.getRank(); i++)
347       for (int i = 1; i < resultTy.getRank(); i++)
372         loc, resultTy.getShape(), resultETy, filteredDims);
388                   loc, resultTy, ValueRange{input, weight, iZpVal, kZpVal},
398                          loc, resultTy, ValueRange{input, weight},
422     ShapedType resultTy = cast<ShapedType>(op->getResult(0).getType());
423     int64_t resultRank = resultTy.getRank();
426     Type resultETy = resultTy.getElementType();
438         loc, input, weight, resultTy, padAttr.asArrayRef(),
454     auto resultShape = resultTy.getShape();
507         loc, resultTy.getShape(), resultETy, filteredDims);
525           loc, resultTy, conv, reassociationMap);
530                   loc, resultTy, ValueRange({bias, convReshape}),
553           loc, resultTy, conv, reassociationMap);
706     TensorType resultTy = op.getType();
717     if (resultTy.isDynamicDim(0))
722       if (!resultTy.isDynamicDim(dim))
742     if (resultTy.isDynamicDim(3))
756     ShapedType resultTy =
758     if (!resultTy)
802         loc, resultTy.getShape(), resultTy.getElementType(), dynamicDims);
813           op, ArrayRef<Type>{resultTy}, ValueRange{paddedInput, fakeWindowDims},
817           op, ArrayRef<Type>{resultTy}, ValueRange{paddedInput, fakeWindowDims},
835     ShapedType resultTy = cast<ShapedType>(op.getType());
839     ShapedType accTy = resultTy.clone(accETy);
898         loc, resultTy.getShape(), resultETy, dynamicDims);
900     auto affineMap = rewriter.getMultiDimIdentityMap(resultTy.getRank());
902         loc, ArrayRef<Type>({resultTy}), ValueRange{poolingOp},
905         getNParallelLoopsAttrs(resultTy.getRank()),