Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_utils.h62 StatusOr<mlir::MemRefType> ConvertTensorShapeToMemRefType(
68 return ConvertTensorShapeToMemRefType(shape, builder); in ConvertTensorShapeToType()
Dhlo_utils.cc83 StatusOr<mlir::MemRefType> ConvertTensorShapeToMemRefType( in ConvertTensorShapeToMemRefType() function
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dmlir_emitter.cc99 TF_ASSIGN_OR_RETURN(mlir::Type ret_memref, ConvertTensorShapeToMemRefType( in EmitMlirFuncAndCall()
105 ConvertTensorShapeToMemRefType(operand_shapes[i], mlir_builder)); in EmitMlirFuncAndCall()