Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 30 for StatusScopedDiagnosticHandler (0.45 sec)

  1. tensorflow/compiler/mlir/lite/experimental/tac/tac_module.cc

                           mlir::OpPassManager::Nesting::Implicit);
      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");
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 08 01:19:25 UTC 2023
    - 5.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tfrt/function/function.cc

      print_flags.elideLargeElementsAttrs();
    
      if (VLOG_IS_ON(1)) {
        VLOG(1) << "Input TF Executor dialect:";
        DumpMlirOpToFile("tf_to_tfrt_tf_executor_dialect", module);
      }
    
      mlir::StatusScopedDiagnosticHandler diag_handler(module.getContext());
    
      // Lower MLIR TF Dialect to MLIR TFRT CoreRT dialect.
      mlir::PassManager pm(module.getContext());
      tensorflow::applyTensorflowAndCLOptions(pm);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 08:13:15 UTC 2024
    - 3.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/sparsity/sparsify_model.cc

    namespace mlir {
    namespace lite {
    
    absl::Status SparsifyModel(const tflite::ModelT& input_model,
                               flatbuffers::FlatBufferBuilder* builder) {
      MLIRContext context;
      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: Mon Jun 10 20:16:40 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.h

        std::unique_ptr<tensorflow::SavedModelBundle>* saved_model_bundle);
    
    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);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 08:30:24 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/api/v1/tf_dialect_to_executor.cc

      PassManager tf_to_executor(module.getContext());
      ::tensorflow::applyTensorflowAndCLOptions(tf_to_executor);
      tf_to_executor.enableVerifier();
    
      AddTfDialectToExecutorPasses(tf_to_executor);
    
      mlir::StatusScopedDiagnosticHandler diag_handler(
          module.getContext(), /*propagate=*/false,
          /*filter_stack=*/!VLOG_IS_ON(1));
    
      if (VLOG_IS_ON(1) ||
          DEBUG_DATA_DUMPER()->ShouldDump(module_name.str(), kDebugGroupMain)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 16 03:41:02 UTC 2024
    - 7.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/tf_to_tfl_flatbuffer.cc

        QuantizationConfig* quantization_config,
        const PyFunctionLibrary* quantization_py_function_lib,
        const SavedModelBundle* saved_model_bundle, mlir::PassManager& pass_manager,
        mlir::StatusScopedDiagnosticHandler& status_handler, ModuleOp& module) {
      // TODO: b/194747383 - We need to valid that indeed the "main" func is
      // presented.
      AddPreQuantizationStableHloToTfPasses(/*entry_function_name=*/"main",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 18:01:23 UTC 2024
    - 23.8K bytes
    - Viewed (0)
  7. 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)
  8. 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));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 6.3K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/api/v2/cluster_tf.cc

      bridge.enableVerifier();
      ::tensorflow::applyTensorflowAndCLOptions(bridge);
    
      // Populate a passmanager with the list of passes that implement the bridge.
      pipeline_builder(bridge);
    
      mlir::StatusScopedDiagnosticHandler diag_handler(
          module.getContext(), /*propagate=*/false,
          /*filter_stack=*/!VLOG_IS_ON(1));
    
      if (VLOG_IS_ON(1) ||
          DEBUG_DATA_DUMPER()->ShouldDump(module_name.str(), kDebugGroupMain)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 28 22:25:18 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tf2xla/api/v1/cluster_tf.cc

      bridge.enableVerifier();
      ::tensorflow::applyTensorflowAndCLOptions(bridge);
    
      // Populate a passmanager with the list of passes that implement the bridge.
      pipeline_builder(bridge);
    
      mlir::StatusScopedDiagnosticHandler diag_handler(
          module.getContext(), /*propagate=*/false,
          /*filter_stack=*/!VLOG_IS_ON(1));
    
      if (VLOG_IS_ON(1) ||
          DEBUG_DATA_DUMPER()->ShouldDump(module_name.str(), kDebugGroupMain)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 28 22:25:18 UTC 2024
    - 9.8K bytes
    - Viewed (0)
Back to top