Searched defs:element_ty (Results 1 – 14 of 14) sorted by relevance
61 Location loc, Type size_ty, Type element_ty, in ConvertArrayAttrToStackAllocatedArray()93 Location loc, Type size_ty, Type element_ty, in ConvertIntegerArrayAttrToStackAllocatedArray()
99 Type element_ty = lhs_type.getElementType(); in IdentityArithmeticOpFolder() local
65 Type element_ty = getElementTypeOrSelf(input_ty); in InferReductionOpType() local
166 Type element_ty = on_value.getType().cast<TensorType>().getElementType(); in InferOneHotOpType() local751 auto element_ty = tensor_ty.getElementType(); in GetReshapeOutputType() local978 Type element_ty = e.getType().cast<TensorType>().getElementType(); in InferSelectV2OpType() local3440 Type element_ty = operand_ty.getElementType(); in inferReturnTypeComponents() local
110 Type element_ty = splat.getType().getElementType(); in fold() local2496 Type element_ty = input.getType().cast<TensorType>().getElementType(); in InferExpandDimsOpType() local
55 mlir::Type element_ty = complex_type.getElementType(); in TypeToPrimitiveType() local
114 Type element_ty = shaped_ty.getElementType(); in ConstantFoldFallbackHook() local
376 auto element_ty = input_ty.getElementType(); in matchAndRewrite() local1063 auto element_ty = input_ty.getElementType(); in matchAndRewrite() local
120 Type element_ty = shaped_ty.getElementType(); in FoldOperation() local
51 auto element_ty = TypeConverter::ConvertElementType(shaped.getElementType()); in VerifyMemrefOperand() local
839 Type element_ty = getElementTypeOrSelf(ty); in DropTypeHelper() local
510 mlir::Type element_ty = getElementTypeOrSelf(type); in GetSplat() local
1759 auto element_ty = input_ty.getElementType(); in GetReshapeOutputType() local
781 auto element_ty = ty.cast<TensorType>().getElementType(); in GetEpsilonValue() local