Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for mlir_dump_file_name_ (0.2 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/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)
  4. 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)
  5. 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