Searched defs:ranked_type (Results 1 – 15 of 15) sorted by relevance
63 if (auto ranked_type = type.dyn_cast<RankedTensorType>()) { in ShuffleRankedTensorType() local
56 auto ranked_type = result_type.dyn_cast<RankedTensorType>(); in DeduceEqualCmpOpType() local
705 auto ranked_type = type.dyn_cast<RankedTensorType>(); in fold() local
114 auto ranked_type = op.input().getType().dyn_cast<RankedTensorType>(); in HandlePad() local422 auto ranked_type = input.getType().dyn_cast<RankedTensorType>(); in HandleHostReplicatedInputs() local462 auto ranked_type = input.value().getType().dyn_cast<RankedTensorType>(); in HandleCluster() local473 auto ranked_type = input.getType().dyn_cast<RankedTensorType>(); in Conv2DInputShapeCanTransform() local
88 if (auto ranked_type = cond.getType().dyn_cast<RankedTensorType>()) in ConvertConditionToBoolean() local
101 if (RankedTensorType ranked_type = type.dyn_cast<RankedTensorType>()) { in VerifySharding() local
253 RankedTensorType ranked_type = type.dyn_cast<RankedTensorType>(); in DropFirstDimension() local
121 if (auto ranked_type = val.getType().dyn_cast<RankedTensorType>()) { in IsDimensionsDegenerateExceptLastOne() local
90 if (auto ranked_type = memref_type.dyn_cast<MemRefType>()) { in EmitMemRefPrint() local
237 if (auto ranked_type = type.dyn_cast<mlir::RankedTensorType>()) { in GetShapeOfValue() local
216 mlir::RankedTensorType ranked_type = in UpdateFunctionArgsUsingLayout() local
89 if (auto ranked_type = t.dyn_cast<RankedTensorType>()) { in GetShapeFromMlirType() local
1346 auto ranked_type = result_type.dyn_cast<mlir::RankedTensorType>(); in ImportInstructionImpl() local
924 auto ranked_type = element_shape.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
757 RankedTensorType ranked_type = tensor_type.dyn_cast<RankedTensorType>(); in ChangeTensorElementType() local2671 if (auto ranked_type = op.getType().template dyn_cast<RankedTensorType>()) in matchAndRewrite() local