Searched refs:operand_ty (Results 1 – 5 of 5) sorted by relevance
40 auto operand_ty = operand.getType().cast<ShapedType>(); in matchAndRewrite() local41 if (!operand_ty.hasRank()) { in matchAndRewrite()98 if (it.value().getSExtValue() != operand_ty.getDimSize(it.index())) { in matchAndRewrite()106 for (auto dim : operand_ty.getShape().drop_front()) { in matchAndRewrite()119 RankedTensorType::get(index_select_shape, operand_ty.getElementType()), in matchAndRewrite()
461 auto operand_ty = (*operands.begin()).getType().cast<ShapedType>(); in inferReturnTypes() local462 Type element_ty = operand_ty.getElementType(); in inferReturnTypes()468 if (operand_ty.hasRank()) { in inferReturnTypes()469 result_ty = RankedTensorType::get(operand_ty.getShape(), element_ty); in inferReturnTypes()517 Type operand_ty = operand.getType(); in build() local518 if (auto ranked_ty = operand_ty.dyn_cast<RankedTensorType>()) { in build()527 auto operand_ty = getOperand().getType().cast<TensorType>(); in fold() local529 if (operand_ty == result_ty) return getOperand(); in fold()536 if (operand_ty.getElementType().isUnsignedInteger() || in fold()1749 static TensorType GetReduceResultType(Type operand_ty, in GetReduceResultType() argument[all …]
1587 auto operand_ty = val.getType().dyn_cast<ShapedType>(); in VerifyStridedSliceBase() local1588 if (!operand_ty || !operand_ty.hasStaticShape()) { in VerifyStridedSliceBase()1595 operand_ty = attr.getType(); in VerifyStridedSliceBase()1598 if (operand_ty.getRank() != 1) in VerifyStridedSliceBase()1602 int64_t length = operand_ty.getDimSize(0); in VerifyStridedSliceBase()
463 auto is_valid_broadcasting = [](ShapedType operand_ty, ShapedType identity_ty,468 if (scalar_identity) return operand_ty == result_ty;475 return operand_ty == result_ty && identity_ty == result_ty &&
2655 auto operand_ty = operand.getType().cast<TensorType>(); in matchAndRewrite() local2656 auto scalar_ty = RankedTensorType::get({}, operand_ty.getElementType()); in matchAndRewrite()