Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils_test.cc29 TEST(ConvertTensorShapeToType, Simple) { in TEST() argument
38 ConvertTensorShapeToType<mlir::RankedTensorType>(shape, builder)); in TEST()
53 ConvertTensorShapeToType<mlir::RankedTensorType>(shape, builder)); in TEST()
Dhlo_utils.h56 static StatusOr<TypeT> ConvertTensorShapeToType(const Shape& shape, in ConvertTensorShapeToType() function
77 inline StatusOr<mlir::MemRefType> ConvertTensorShapeToType( in ConvertTensorShapeToType() function
105 return ConvertTensorShapeToType<TypeT>(shape, builder); in ConvertShapeToType()
Dhlo_utils.cc108 ConvertTensorShapeToType<mlir::RankedTensorType>( in CreateDenseElementsAttrFromLiteral()