Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for SaveQuantizationReportInstrumentation (0.56 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/instrumentations/save_report.cc

                   << " with status: " << save_status;
      }
    }
    
    }  // namespace
    
    SaveQuantizationReportInstrumentation::SaveQuantizationReportInstrumentation(
        std::optional<absl::string_view> file_path)
        : file_path_(OptionalStringViewToOptionalString(file_path)) {}
    
    void SaveQuantizationReportInstrumentation::runAfterPass(Pass* pass,
                                                             Operation* op) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 02:59:01 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/instrumentations/save_report_test.cc

      QuantizeCompositeFunctionsPassOptions options;
      pm.addPass(createQuantizeCompositeFunctionsPass(options));
      pm.addInstrumentation(std::make_unique<SaveQuantizationReportInstrumentation>(
          /*file_path=*/std::nullopt));
    
      // The report file is not created and `SaveQuantizationReportInstrumentation`
      // is not run, but the passes still run without errors.
      const LogicalResult run_result = pm.run(*module_op);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 02:59:01 UTC 2024
    - 9.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/stablehlo/instrumentations/save_report.h

    // should exist (this pass instrumentation will not `mkdir` them).
    //
    // See `QuantizationReport` for further details on the quantization report.
    class SaveQuantizationReportInstrumentation : public PassInstrumentation {
     public:
      // `file_path` is the path to save the report file. The report file is in
      // textproto format so a `.txtpb` extension is preferred but it doesn't result
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 02:59:01 UTC 2024
    - 2.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/cc/post_calibration.cc

          kName, /*add_passes_func=*/
          [&config](PassManager& pm) {
            // Add instrumentation to save quantization report after quantization.
            pm.addInstrumentation(
                std::make_unique<SaveQuantizationReportInstrumentation>(
                    GetReportFilePath(config)));
    
            AddPostCalibrationPasses(pm, config.pipeline_config(), config.specs());
          },
          *ctx_, module_op));
      return module_op;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 02:59:01 UTC 2024
    - 2.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/stablehlo/cc/weight_only_ptq.cc

          kName, /*add_passes_func=*/
          [&config](PassManager& pm) {
            // Add instrumentation to save quantization report after quantization.
            pm.addInstrumentation(
                std::make_unique<SaveQuantizationReportInstrumentation>(
                    GetReportFilePath(config)));
    
            AddWeightOnlyQuantizationPasses(pm, config.specs(),
                                            config.pipeline_config(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 02:59:01 UTC 2024
    - 5.1K bytes
    - Viewed (0)
Back to top