Home
last modified time | relevance | path

Searched refs:ConvertPrimitiveTypeToMLIRType (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dhlo_utils.h46 StatusOr<mlir::Type> ConvertPrimitiveTypeToMLIRType(PrimitiveType element_type,
57 ConvertPrimitiveTypeToMLIRType(xla_ty.element_type(), builder); in ConvertTensorShapeToType()
H A Dhlo_utils.cc93 ConvertPrimitiveTypeToMLIRType(shape.element_type(), builder); in ConvertTensorShapeToMemRefType()
228 StatusOr<mlir::Type> ConvertPrimitiveTypeToMLIRType(PrimitiveType element_type, in ConvertPrimitiveTypeToMLIRType() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dxla_call_module_op.cc166 ConvertPrimitiveTypeToMLIRType(xla_shape.element_type(), builder)); in RefineDynamicShapes()