Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for StatusScopedDiagnosticHandler (0.44 sec)

  1. 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)
  2. 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)
  3. 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)
  4. tensorflow/compiler/mlir/tf2xla/api/v1/compile_tf_graph.cc

      manager.addNestedPass<mlir::func::FuncOp>(
          mlir::CreateFunctionalToExecutorDialectConversionPass());
      manager.addPass(mlir::CreateBreakUpIslandsPass());
    
      mlir::StatusScopedDiagnosticHandler diag_handler(module.getContext());
    
      if (VLOG_IS_ON(2)) {
        llvm::StringRef module_name = llvm::StringRef();
        constexpr const char* kDebugGroupBridgePhase2 =
            "v1_prepare_and_export_to_library";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 22:19:26 UTC 2024
    - 14K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

        // If `allow_tensorlist_pass_through` is set to true, if legalization fails
        // we should not leak the diagnostic info outside this pass. Hence we use
        // a `StatusScopedDiagnosticHandler` here to capture diagnostics generated
        // within this pass.
        StatusScopedDiagnosticHandler handler(context);
        if (failed(applyPartialConversion(module, target, std::move(patterns)))) {
          auto _ = handler.ConsumeStatus();
        }
      }
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
Back to top