Searched defs:element_ty (Results 1 – 12 of 12) sorted by relevance
57 Type element_ty = getElementTypeOrSelf(ty); in getSplat() local
55 mlir::Type element_ty = complex_type.getElementType(); in TypeToPrimitiveType() local
109 Type element_ty = shaped_ty.getElementType(); in ConstantFoldFallbackHook() local
386 auto element_ty = input_ty.getElementType(); in matchAndRewrite() local1073 auto element_ty = input_ty.getElementType(); in matchAndRewrite() local
98 Location loc, Type size_ty, Type element_ty, in ConvertArrayAttrToStackAllocatedArray()130 Location loc, Type size_ty, Type element_ty, in ConvertIntegerArrayAttrToStackAllocatedArray()148 Type element_ty = LLVM::LLVMPointerType::get(rewriter->getI8Type()); in ConvertStrArrayAttrToStackAllocatedArray() local
150 Type element_ty = on_value.getType().cast<TensorType>().getElementType(); in InferOneHotOpType() local712 auto element_ty = tensor_ty.getElementType(); in GetReshapeOutputType() local937 Type element_ty = e.getType().cast<TensorType>().getElementType(); in InferSelectV2OpType() local3253 Type element_ty = operand_ty.getElementType(); in inferReturnTypes() local
2328 Type element_ty = input.getType().cast<TensorType>().getElementType(); in InferExpandDimsOpType() local
776 Type element_ty = getElementTypeOrSelf(ty); in DropTypeHelper() local
717 Type element_ty = operand_ty.getElementType(); in inferReturnTypes() local1274 auto element_ty = ComplexType::get(getElementTypeOrSelf(type)); in inferReturnTypes() local1303 auto element_ty = getElementTypeOrSelf(type); in CreateRealType() local2317 Type element_ty = getElementTypeOrSelf(operand_ty); in GetReduceResultType() local
504 mlir::Type element_ty = getElementTypeOrSelf(type); in GetSplat() local
1614 auto element_ty = input_ty.getElementType(); in GetReshapeOutputType() local
856 auto element_ty = ty.cast<TensorType>().getElementType(); in GetEpsilonValue() local