- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for status_handler (0.24 sec)
-
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc
return status_handler.ConsumeStatus(); } if (export_to_mlir) { llvm::raw_string_ostream os(*result); module.print(os); return status_handler.ConsumeStatus(); } pass_manager.clear(); pass_manager.addPass(mlir::odml::createLegalizeStablehloToVhloPass()); if (failed(pass_manager.run(module))) { return status_handler.Combine(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/dump_graph.cc
flib_def ? *flib_def : graph.flib_def())); break; } } if (failed(mlir::verify(*module))) { return status_handler.ConsumeStatus(); } return status_handler.ConsumeStatus(); }; TF_RETURN_IF_ERROR(convert()); module->print(os, config.op_printing_flags); return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 03:47:51 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/run_passes.cc
namespace quantization { absl::Status RunPassesOnModuleOp( std::optional<absl::string_view> mlir_dump_file_name, mlir::PassManager& pass_manager, mlir::ModuleOp module_op) { mlir::StatusScopedDiagnosticHandler statusHandler(module_op.getContext(), /*propagate=*/true); absl::StatusOr<std::unique_ptr<llvm::raw_ostream>> dump_file; if (mlir_dump_file_name) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 07 01:17:26 UTC 2024 - 1.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/sparsity/sparsify_model.cc
pm.addPass(TFL::CreateDenseToSparsePass()); if (failed(pm.run(module.get()))) { LOG(ERROR) << "Failed to sparsify: " << statusHandler.ConsumeStatus().message(); return absl::InternalError(absl::StrCat( "Failed to sparsify: ", statusHandler.ConsumeStatus().message())); } // Export the results to the builder std::string result; tflite::FlatbufferExportOptions options;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:16:40 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_model.cc
denylisted_mlir_op_names.insert(TfLiteToMlir(entry)); } DialectRegistry registry; registry.insert<mlir::TFL::TensorFlowLiteDialect>(); MLIRContext context(registry); StatusScopedDiagnosticHandler statusHandler(&context, /*propagate=*/true); OwningOpRef<mlir::ModuleOp> module = tflite::FlatBufferToMlir( model_buffer, &context, UnknownLoc::get(&context)); if (!module) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 6.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/lite/quantize_weights.cc
for (auto& entry : denylisted_ops) { denylisted_mlir_op_names.insert(TfLiteToMlir(entry)); } DialectRegistry registry; MLIRContext context(registry); StatusScopedDiagnosticHandler statusHandler(&context, /*propagate=*/true); // Import input_model to a MLIR module flatbuffers::FlatBufferBuilder input_builder;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 23:15:24 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/quantize_preprocess.cc
const bool deserialize_xla_call_module, llvm::ArrayRef<llvm::ArrayRef<int64_t>> input_arg_shapes) { mlir::PassManager pm_before_freezing_variables(context); mlir::StatusScopedDiagnosticHandler statusHandler(module_op.getContext(), /*propagate=*/true); mlir::TF::StandardPipelineOptions standard_pipeline_options; standard_pipeline_options.enable_inliner = false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 9.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/tac/tac_module.cc
AddTACPass(&pm, options_.hardware_backends); if (!debug_mode) { AddExportTFLPass(&pm, options_.enable_inliner); } mlir::StatusScopedDiagnosticHandler statusHandler(module->getContext(), /*propagate=*/true); if (failed(pm.run(*module))) { return absl::InternalError("conversion error"); } return absl::OkStatus(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 5.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.h
Status ConvertTFExecutorToStablehloFlatbuffer( mlir::PassManager& pass_manager, mlir::ModuleOp module, bool export_to_mlir, mlir::StatusScopedDiagnosticHandler& statusHandler, const toco::TocoFlags& toco_flags, const mlir::TFL::PassConfig& pass_config, std::optional<tensorflow::Session*> session, std::string* result); // Taking a MLIR module in TF executor dialect and a set of parameters,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 08:30:24 UTC 2024 - 4.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/python/mlir.cc
("Invalid pass_pipeline: " + error_stream.str()).c_str()); return "// error"; } mlir::StatusScopedDiagnosticHandler statusHandler(module.getContext()); if (failed(pm.run(module))) { tsl::Set_TF_Status_from_Status(status, statusHandler.ConsumeStatus()); return "// error"; } } return MlirModuleToString(module, show_debug_info); } } // anonymous namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:16:49 UTC 2024 - 19.3K bytes - Viewed (0)