Searched defs:element_ty (Results 1 – 11 of 11) sorted by relevance
41 Type element_ty = getElementTypeOrSelf(ty); in getSplat() local
55 mlir::Type element_ty = complex_type.getElementType(); in TypeToPrimitiveType() local
99 Type element_ty = shaped_ty.getElementType(); in ConstantFoldFallbackHook() local
342 auto element_ty = input_ty.getElementType(); in matchAndRewrite() local1035 auto element_ty = input_ty.getElementType(); in matchAndRewrite() local
377 Type element_ty = getElementTypeOrSelf(ty); in DropTypeHelper() local
149 Type element_ty = on_value.getType().cast<TensorType>().getElementType(); in InferOneHotOpType() local581 auto element_ty = tensor_ty.getElementType(); in GetReshapeOutputType() local811 Type element_ty = e.getType().cast<TensorType>().getElementType(); in InferSelectV2OpType() local
2127 Type element_ty = input.getType().cast<TensorType>().getElementType(); in InferExpandDimsOpType() local
462 Type element_ty = operand_ty.getElementType(); in inferReturnTypes() local964 auto element_ty = ComplexType::get(getElementTypeOrSelf(type)); in inferReturnTypes() local993 auto element_ty = getElementTypeOrSelf(type); in CreateRealType() local1752 Type element_ty = getElementTypeOrSelf(operand_ty); in GetReduceResultType() local
476 mlir::Type element_ty = getElementTypeOrSelf(type); in GetSplat() local
1353 auto element_ty = input_ty.getElementType(); in GetReshapeOutputType() local
908 auto element_ty = ty.cast<TensorType>().getElementType(); in GetEpsilonValue() local