Searched refs:rankedTensor (Results 1 – 4 of 4) sorted by relevance
61 Value rankedTensor) { in createDynamicDimValues() argument62 auto tensorTy = rankedTensor.getType().cast<RankedTensorType>(); in createDynamicDimValues()67 b.create<tensor::DimOp>(loc, rankedTensor, en.index())); in createDynamicDimValues()
34 Value rankedTensor);
335 Value rankedTensor = std::get<0>(it); in splitReductionByScaling() local336 auto t = rankedTensor.getType().cast<RankedTensorType>(); in splitReductionByScaling()340 tensor::createDynamicDimValues(b, loc, rankedTensor); in splitReductionByScaling()
730 MlirType rankedTensor = mlirRankedTensorTypeGet( in printBuiltinTypes() local732 if (!mlirTypeIsATensor(rankedTensor) || in printBuiltinTypes()733 !mlirTypeIsARankedTensor(rankedTensor) || in printBuiltinTypes()734 !mlirAttributeIsNull(mlirRankedTensorTypeGetEncoding(rankedTensor))) in printBuiltinTypes()736 mlirTypeDump(rankedTensor); in printBuiltinTypes()