Home
last modified time | relevance | path

Searched refs:ConvertShapeToType (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc101 TF_ASSIGN_OR_RETURN(mlir::Type ty, ConvertShapeToType<mlir::RankedTensorType>( in ConvGeneralDilatedInternal()
122 TF_ASSIGN_OR_RETURN(mlir::Type ty, ConvertShapeToType<mlir::RankedTensorType>( in FftInternal()
142 TF_ASSIGN_OR_RETURN(mlir::Type ty, ConvertShapeToType<mlir::RankedTensorType>( in CustomCallInternal()
175 TF_ASSIGN_OR_RETURN(mlir::Type ty, ConvertShapeToType<mlir::RankedTensorType>( in ReduceWindowInternal()
205 ConvertShapeToType<mlir::RankedTensorType>(shape, builder_)); in Iota()
215 TF_ASSIGN_OR_RETURN(mlir::Type ty, ConvertShapeToType<mlir::RankedTensorType>( in BitcastConvertTypeInternal()
224 TF_ASSIGN_OR_RETURN(mlir::Type ty, ConvertShapeToType<mlir::RankedTensorType>( in TransposeInternal()
233 TF_ASSIGN_OR_RETURN(mlir::Type ty, ConvertShapeToType<mlir::RankedTensorType>( in RevInternal()
244 TF_ASSIGN_OR_RETURN(mlir::Type ty, ConvertShapeToType<mlir::RankedTensorType>( in SortInternal()
268 TF_ASSIGN_OR_RETURN(mlir::Type ty, ConvertShapeToType<mlir::RankedTensorType>( in WhileInternal()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.h72 static StatusOr<mlir::Type> ConvertShapeToType(const Shape& shape, in ConvertShapeToType() function
79 ConvertShapeToType<TypeT>(subtype, builder)); in ConvertShapeToType()
Dtype_to_shape_test.cc189 ConvertShapeToType<MemRefType>(shape, builder); in TEST()
206 ConvertShapeToType<MemRefType>(shape, builder); in TEST()
Dhlo_function_importer.cc230 TF_ASSIGN_OR_RETURN(auto result_type, ConvertShapeToType<RankedTensorType>( in ImportInstructionImpl()
880 TF_ASSIGN_OR_RETURN(auto ret_type, ConvertShapeToType<RankedTensorType>( in GetMlirTypes()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc900 TF_ASSIGN_OR_RETURN(Type type, xla::ConvertShapeToType<MemRefType>( in EmitConstant()
1350 TF_ASSIGN_OR_RETURN(Type out_type, xla::ConvertShapeToType<MemRefType>( in GetOrCreateArrayView()
1373 xla::ConvertShapeToType<MemRefType>(physical_shape, builder_)); in GetOrCreateArrayView()
1491 TF_ASSIGN_OR_RETURN(auto arg_type, xla::ConvertShapeToType<MemRefType>( in Initialize()