Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 27 for StatusScopedDiagnosticHandler (0.44 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/lower_cluster_to_runtime_ops.cc

      } else {
        AddNonTPULowerClusterToRuntimeOpsPassPipeline(runtime_lowering,
                                                      module_name);
      }
    
      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: Wed Apr 17 18:52:57 UTC 2024
    - 9.4K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_quantized_functions.cc

      PassManager pm(context);
      pm.addPass(createInlinerPass());
      pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());
      pm.addNestedPass<func::FuncOp>(createCSEPass());
    
      StatusScopedDiagnosticHandler diagnostic_handler(context);
      if (failed(pm.run(*module_ref))) {
        emitError(module.getLoc()) << "failed to apply the optimization: "
                                   << diagnostic_handler.ConsumeStatus().message();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 8.7K bytes
    - Viewed (0)
  10. 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;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 12:49:45 UTC 2024
    - 9.8K bytes
    - Viewed (0)
Back to top