Home
last modified time | relevance | path

Searched refs:ConvertTensorShapeToType (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dhlo_utils_test.cc28 TEST(ConvertTensorShapeToType, Simple) { in TEST() argument
38 ConvertTensorShapeToType<mlir::RankedTensorType>(shape, builder)); in TEST()
52 ConvertTensorShapeToType<mlir::RankedTensorType>(shape, builder)); in TEST()
H A Dhlo_utils.h54 static StatusOr<TypeT> ConvertTensorShapeToType(const Shape& xla_ty, in ConvertTensorShapeToType() function
85 inline StatusOr<mlir::MemRefType> ConvertTensorShapeToType( in ConvertTensorShapeToType() function
111 return ConvertTensorShapeToType<TypeT>(shape, builder); in ConvertShapeToType()
H A Dhlo_utils.cc108 ConvertTensorShapeToType<mlir::RankedTensorType>( in CreateDenseElementsAttrFromLiteral()
H A Dhlo_function_importer.cc1360 ConvertTensorShapeToType<mlir::RankedTensorType>( in ImportInstructionImpl()