Home
last modified time | relevance | path

Searched refs:StatusScopedDiagnosticHandler (Results 1 – 18 of 18) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Derror_util.cc22 StatusScopedDiagnosticHandler::StatusScopedDiagnosticHandler( in StatusScopedDiagnosticHandler() function in mlir::StatusScopedDiagnosticHandler
30 StatusScopedDiagnosticHandler::~StatusScopedDiagnosticHandler() { in ~StatusScopedDiagnosticHandler()
37 bool StatusScopedDiagnosticHandler::ok() const { return diag_str_.empty(); } in ok()
39 Status StatusScopedDiagnosticHandler::ConsumeStatus() { in ConsumeStatus()
49 Status StatusScopedDiagnosticHandler::Combine(Status status) { in Combine()
65 LogicalResult StatusScopedDiagnosticHandler::handler(Diagnostic* diag) { in handler()
Derror_util_test.cc30 TEST(ErrorUtilTest, StatusScopedDiagnosticHandler) { in TEST() argument
37 TF_ASSERT_OK(StatusScopedDiagnosticHandler(&context).Combine(Status::OK())); in TEST()
42 StatusScopedDiagnosticHandler handler(&context); in TEST()
51 StatusScopedDiagnosticHandler(&context).Combine(err))); in TEST()
61 StatusScopedDiagnosticHandler ssdh(&context); in TEST()
Derror_util.h37 class StatusScopedDiagnosticHandler : public SourceMgrDiagnosticHandler {
42 explicit StatusScopedDiagnosticHandler(MLIRContext* context,
45 ~StatusScopedDiagnosticHandler();
Dserialize_mlir_module_utils.cc45 mlir::StatusScopedDiagnosticHandler error_handler(mlir_context); in DeserializeMlirModule()
Ddump_graph.cc74 mlir::StatusScopedDiagnosticHandler status_handler(&context); in DumpTextualIRToFile()
Dcompile_mlir_util.cc267 mlir::StatusScopedDiagnosticHandler error_handler(module.getContext()); in RefineShapes()
362 mlir::StatusScopedDiagnosticHandler error_handler(module_op.getContext()); in LegalizeToHlo()
608 mlir::StatusScopedDiagnosticHandler diag_handler(module_op.getContext()); in CompileGraphSetup()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_to_mlir_hlo.cc27 mlir::StatusScopedDiagnosticHandler diag_handler(module.getContext()); in ConvertHloToMlirHlo()
34 mlir::StatusScopedDiagnosticHandler diag_handler(module.getContext()); in ConvertHloToMlirHlo()
Dmlir_hlo_to_hlo.cc1776 mlir::StatusScopedDiagnosticHandler diag_handler(module.getContext()); in ConvertMlirHloToHlo()
1807 mlir::StatusScopedDiagnosticHandler diag_handler(module.getContext()); in BuildHloFromMlirHlo()
/external/tensorflow/tensorflow/compiler/mlir/python/
Dmlir.cc63 mlir::StatusScopedDiagnosticHandler statusHandler(module.getContext()); in RunPassPipelineOnModule()
229 mlir::StatusScopedDiagnosticHandler diagnostic_handler(&context); in ExperimentalConvertSavedModelV1ToMlir()
246 mlir::StatusScopedDiagnosticHandler diagnostic_handler(&context); in ExperimentalRunPassPipeline()
265 mlir::StatusScopedDiagnosticHandler diagnostic_handler(&context); in ExperimentalRunPassPipeline()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dbridge.cc75 mlir::StatusScopedDiagnosticHandler diag_handler(module.getContext()); in RunTPUBridge()
179 mlir::StatusScopedDiagnosticHandler diag_handler(module.getContext()); in RunBridgeWithStandardPipeline()
Dgraph_optimization_pass.cc63 StatusScopedDiagnosticHandler diag_handler(module.getContext()); in Run()
/external/tensorflow/tensorflow/compiler/mlir/lite/sparsity/
Dsparsify_model.cc40 StatusScopedDiagnosticHandler statusHandler(&context, in SparsifyModel()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dmlir_roundtrip_pass.cc44 mlir::StatusScopedDiagnosticHandler status_handler(context); in Import()
Dimport_model.cc423 mlir::StatusScopedDiagnosticHandler error_handler_;
3836 mlir::StatusScopedDiagnosticHandler diag_handler(module.getContext()); in LiftVariables()
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
Dquantize_model.cc59 StatusScopedDiagnosticHandler statusHandler(&context, in QuantizeModel()
/external/tensorflow/tensorflow/compiler/mlir/tfjs/translate/
Dtf_to_tfjs_json.cc105 mlir::StatusScopedDiagnosticHandler statusHandler(module.getContext(), in ConvertTFOpsToTfjsJSON()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_to_tfl_flatbuffer.cc159 mlir::StatusScopedDiagnosticHandler statusHandler(module.getContext(), in ConvertTFExecutorToTFLOrFlatbuffer()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc531 StatusScopedDiagnosticHandler diag_handler(func_.getContext()); in GetFunctionDef()