Searched refs:ranked_ty (Results 1 – 7 of 7) sorted by relevance
105 auto ranked_ty = input_ty.dyn_cast<RankedTensorType>();106 if (!ranked_ty) return UnrankedTensorType::get(element_ty);107 int64_t rank = ranked_ty.getRank();132 ArrayRef<int64_t> shape = ranked_ty.getShape();164 RankedTensorType ranked_ty = ty.dyn_cast<RankedTensorType>();165 if (!ranked_ty) continue;167 int64_t rank = ranked_ty.getRank();173 << "operand type " << ranked_ty181 int64_t dim = ranked_ty.getDimSize(i);196 return op->emitError() << "operand type " << ranked_ty
382 if (auto ranked_ty = ty.dyn_cast<RankedTensorType>()) { in DropTypeHelper() local383 return RankedTensorType::get(ranked_ty.getShape(), default_ty); in DropTypeHelper()
890 auto ranked_ty = input_ty.dyn_cast<RankedTensorType>(); in ConvertShapeToAttr() local891 if (!ranked_ty || !ranked_ty.hasStaticShape()) return {}; in ConvertShapeToAttr()893 auto shape = ranked_ty.getShape(); in ConvertShapeToAttr()
518 if (auto ranked_ty = operand_ty.dyn_cast<RankedTensorType>()) { in build() local519 result_ty = RankedTensorType::get(ranked_ty.getShape(), result_element_ty); in build()1754 auto ranked_ty = operand_ty.dyn_cast<RankedTensorType>(); in GetReduceResultType() local1755 if (!ranked_ty) return UnrankedTensorType::get(element_ty); in GetReduceResultType()1757 int64_t rank = ranked_ty.getRank(); in GetReduceResultType()1763 if (!dims_mask[i]) shape.push_back(ranked_ty.getDimSize(i)); in GetReduceResultType()2370 RankedTensorType ranked_ty = ty.dyn_cast<RankedTensorType>(); in inferReturnTypes() local2371 if (!ranked_ty) { in inferReturnTypes()2385 int64_t rank = ranked_ty.getRank(); in inferReturnTypes()2413 shape.push_back(InferSliceDim(ranked_ty.getDimSize(i), start[i], limit[i], in inferReturnTypes()[all …]
406 auto ranked_ty = ty.dyn_cast<ShapedType>(); in LegalizeOp() local407 if (!ranked_ty || !ranked_ty.hasStaticShape()) { in LegalizeOp()
253 RankedTensorType ranked_ty = ty.dyn_cast<RankedTensorType>(); in GetDimSize() local254 if (!ranked_ty) return -1; in GetDimSize()256 return ranked_ty.getDimSize(index); in GetDimSize()
117 auto ranked_ty = ty.dyn_cast<RankedTensorType>(); in InferExpandDimsType() local120 if (!ranked_ty) return ty; in InferExpandDimsType()122 auto shape = llvm::to_vector<4>(ranked_ty.getShape()); in InferExpandDimsType()123 if (axis < 0) axis += ranked_ty.getRank() + 1; in InferExpandDimsType()126 return RankedTensorType::get(shape, ranked_ty.getElementType()); in InferExpandDimsType()