Searched refs:ranked_type (Results 1 – 11 of 11) sorted by relevance
159 auto ranked_type = op.input().getType().dyn_cast<RankedTensorType>(); in HandlePad() local160 if (!ranked_type) return failure(); in HandlePad()161 auto pad_input_shape = ranked_type.getShape(); in HandlePad()467 auto ranked_type = input.getType().dyn_cast<RankedTensorType>(); in HandleHostReplicatedInputs() local468 if (!ranked_type) return false; in HandleHostReplicatedInputs()469 auto input_shape = ranked_type.getShape(); in HandleHostReplicatedInputs()507 auto ranked_type = input.value().getType().dyn_cast<RankedTensorType>(); in HandleCluster() local508 if (!ranked_type) continue; in HandleCluster()509 auto input_shape = ranked_type.getShape(); in HandleCluster()518 auto ranked_type = input.getType().dyn_cast<RankedTensorType>(); in Conv2DInputShapeCanTransform() local[all …]
86 if (auto ranked_type = cond.getType().dyn_cast<RankedTensorType>()) in ConvertConditionToBoolean() local87 if (ranked_type.getRank() == 0 && in ConvertConditionToBoolean()88 ranked_type.getElementType().isSignlessInteger(1)) in ConvertConditionToBoolean()
157 RankedTensorType ranked_type = type.dyn_cast<RankedTensorType>(); in DropFirstDimension() local158 if (!ranked_type) return {}; in DropFirstDimension()160 ranked_type.getShape().drop_front(); in DropFirstDimension()161 return RankedTensorType::get(dims_except_first, ranked_type.getElementType()); in DropFirstDimension()
118 if (auto ranked_type = val.getType().dyn_cast<RankedTensorType>()) { in IsDimensionsDegenerateExceptLastOne() local119 return IsDimensionsDegenerateExceptLastOne(ranked_type.getShape()); in IsDimensionsDegenerateExceptLastOne()
84 auto ranked_type = result_type.dyn_cast<RankedTensorType>();85 if (!ranked_type) return UnrankedTensorType::get(builder->getI1Type());87 return RankedTensorType::get(ranked_type.getShape(), builder->getI1Type());330 if (auto ranked_type = type.dyn_cast<RankedTensorType>()) {331 ArrayRef<int64_t> shape = ranked_type.getShape();338 return RankedTensorType::get(new_shape, ranked_type.getElementType());
533 auto ranked_type = type.dyn_cast<RankedTensorType>(); in fold() local534 if (!ranked_type) return {}; in fold()541 int32_t rank = ranked_type.getRank(); in fold()
84 if (auto ranked_type = t.dyn_cast<RankedTensorType>()) { in GetShapeFromMlirType() local85 return tensorflow::PartialTensorShape(ranked_type.getShape()); in GetShapeFromMlirType()
228 if (auto ranked_type = op.getType().dyn_cast<RankedTensorType>()) { in matchAndRewrite() local230 MemRefType::get(ranked_type.getShape(), ranked_type.getElementType()); in matchAndRewrite()
761 auto ranked_type = element_shape.getType().dyn_cast<RankedTensorType>(); in matchAndRewrite() local763 if ((ranked_type && ranked_type.getRank() == 0) || in matchAndRewrite()
966 if (auto ranked_type = type.dyn_cast<RankedTensorType>()) { in inferReturnTypes() local967 result_ty = RankedTensorType::get(ranked_type.getShape(), element_ty); in inferReturnTypes()998 if (auto ranked_type = type.dyn_cast<RankedTensorType>()) { in CreateRealType() local999 return RankedTensorType::get(ranked_type.getShape(), element_ty); in CreateRealType()
884 RankedTensorType ranked_type = tensor_type.dyn_cast<RankedTensorType>(); in ChangeTensorElementType() local885 if (ranked_type) { in ChangeTensorElementType()886 return RankedTensorType::get(ranked_type.getShape(), element_type); in ChangeTensorElementType()2273 if (auto ranked_type = op.getType().template dyn_cast<RankedTensorType>()) in matchAndRewrite() local2275 RankedTensorType::get(ranked_type.getShape(), sum_element_type); in matchAndRewrite()