Searched defs:ranked_type (Results 1 – 10 of 10) sorted by relevance
159 auto ranked_type = op.input().getType().dyn_cast<RankedTensorType>(); in HandlePad() local467 auto ranked_type = input.getType().dyn_cast<RankedTensorType>(); in HandleHostReplicatedInputs() local507 auto ranked_type = input.value().getType().dyn_cast<RankedTensorType>(); in HandleCluster() local518 auto ranked_type = input.getType().dyn_cast<RankedTensorType>(); in Conv2DInputShapeCanTransform() local
86 if (auto ranked_type = cond.getType().dyn_cast<RankedTensorType>()) in ConvertConditionToBoolean() local
157 RankedTensorType ranked_type = type.dyn_cast<RankedTensorType>(); in DropFirstDimension() local
118 if (auto ranked_type = val.getType().dyn_cast<RankedTensorType>()) { in IsDimensionsDegenerateExceptLastOne() local
84 if (auto ranked_type = t.dyn_cast<RankedTensorType>()) { in GetShapeFromMlirType() local
228 if (auto ranked_type = op.getType().dyn_cast<RankedTensorType>()) { in matchAndRewrite() local
761 auto ranked_type = element_shape.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local
966 if (auto ranked_type = type.dyn_cast<RankedTensorType>()) { in inferReturnTypes() local998 if (auto ranked_type = type.dyn_cast<RankedTensorType>()) { in CreateRealType() local
533 auto ranked_type = type.dyn_cast<RankedTensorType>(); in fold() local
884 RankedTensorType ranked_type = tensor_type.dyn_cast<RankedTensorType>(); in ChangeTensorElementType() local2273 if (auto ranked_type = op.getType().template dyn_cast<RankedTensorType>()) in matchAndRewrite() local