Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/
H A Dhlo_utils.h81 StatusOr<mlir::MemRefType> ConvertTensorShapeToMemRefType(
91 return ConvertTensorShapeToMemRefType(shape, builder); in ConvertTensorShapeToType()
H A Dhlo_utils.cc90 StatusOr<mlir::MemRefType> ConvertTensorShapeToMemRefType( in ConvertTensorShapeToMemRefType() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dmlir_emitter.cc102 TF_ASSIGN_OR_RETURN(mlir::Type ret_memref, ConvertTensorShapeToMemRefType( in EmitMlirFuncAndCall()
108 ConvertTensorShapeToMemRefType(operand_shapes[i], mlir_builder)); in EmitMlirFuncAndCall()