Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for mlir_dump_file_name_ (0.25 sec)

  1. tensorflow/compiler/mlir/quantization/stablehlo/cc/pass_pipeline.cc

      // replaced by individual `Method` in `QuantizationSpecs`.
      options.enable_per_channel_quantized_weight_ = true;
      // For debugging purposes.
      options.mlir_dump_file_name_ = "quantize_composite_functions";
      options.merge_fusion_with_dequantize_ =
          pipeline_config.merge_fusion_with_dequantize();
    
      AddShapeLegalizationPasses(pm);
      pm.addNestedPass<func::FuncOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/stablehlo/passes/quantize_composite_functions.cc

      if (merge_fusion_with_dequantize_) {
        pm.addPass(createMergeFusionWithDequantizePass());
      }
    
      ModuleOp module_op = getOperation();
      if (const absl::Status pm_run_status =
              RunPassesOnModuleOp(mlir_dump_file_name_, pm, module_op);
          !pm_run_status.ok()) {
        signalPassFailure();
      }
    }
    
    }  // namespace
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 03 02:59:01 UTC 2024
    - 4.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc

          const int min_num_elements_for_weights,
          const bool enable_legacy_weight_only,
          std::optional<const std::string> mlir_dump_file_name)
          : enable_legacy_weight_only_(enable_legacy_weight_only),
            min_num_elements_for_weights_(min_num_elements_for_weights),
            mlir_dump_file_name_(std::move(mlir_dump_file_name)) {
        quantization_method_ = quantization_method;
        target_opset_ = target_opset;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 54.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/quantization/stablehlo/passes/passes.td

        Option<"enable_per_channel_quantized_weight_",
            "enable-per-channel-quantized-weight",
            "bool", /*default=*/"true",
            "Whether to enable per-channel quantized weights.">,
        Option<"mlir_dump_file_name_", "mlir-dump-file-name",
            "std::optional<std::string>", /*default=*/"std::nullopt",
            "MLIR dump file name.">,
        Option<"merge_fusion_with_dequantize_",
            "merge-fusion-with-dequantize",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 06:31:57 UTC 2024
    - 10.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/cc/run_passes.cc

    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)
  6. tensorflow/compiler/mlir/quantization/tensorflow/quantize_preprocess.cc

        AddXlaCallModuleOpDeserializationPasses(pm_after_freezing_variables);
      }
    
      if (const auto pre_variable_freezing_status = RunPassesOnModuleOp(
              /*mlir_dump_file_name=*/absl::StrCat(
                  mlir_dump_file_prefix, "_preprocess_pre_variable_freezing"),
              pm_before_freezing_variables, module_op);
          !pre_variable_freezing_status.ok()) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 12:49:45 UTC 2024
    - 9.8K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/tensorflow/cc/run_passes.h

    // conditions.
    //
    // Returns a non-OK status when the pass run fails or it fails to create an MLIR
    // dump file.
    absl::Status RunPassesOnModuleOp(
        std::optional<absl::string_view> mlir_dump_file_name,
        mlir::PassManager& pass_manager, mlir::ModuleOp module_op);
    
    }  // namespace quantization
    }  // namespace tensorflow
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 07 01:17:26 UTC 2024
    - 3K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/quantization/stablehlo/passes/testing/test_tf_to_stablehlo_pass.cc

      ModuleOp module_op = getOperation();
      MLIRContext* ctx = &getContext();
      mlir::PassManager pm(ctx);
    
      AddTFToStablehloPasses(pm);
      if (!RunPassesOnModuleOp(
               /*mlir_dump_file_name=*/"test_tf_to_stablehlo_pass", pm, module_op)
               .ok()) {
        return signalPassFailure();
      }
    }
    
    }  // namespace
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 17:19:36 UTC 2024
    - 3.1K bytes
    - Viewed (0)
Back to top