Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.h85 Status RefineShapes(llvm::ArrayRef<TensorOrResourceShape> arg_shapes,
Dcompile_mlir_util.cc228 Status RefineShapes(llvm::ArrayRef<TensorOrResourceShape> arg_shapes, in RefineShapes() function
410 TF_RETURN_IF_ERROR(RefineShapes(arg_shapes, module_op)); in CompileMlirSetup()
Dtf_xla_mlir_translate.cc248 TF_RETURN_IF_ERROR(RefineShapes(arg_shapes, module_op)); in CompileMlirToXlaHloViaBuilder()