Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for AddProcessNchwTensorPasses (0.15 sec)

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

      // Convert NCHW tensors to NHWC at along with extra optimizations as
      // downstream passes perform better optimizations when dealing with NHWC
      // formatted tensors.
      AddProcessNchwTensorPasses(pm);
    
      pm.addPass(CreateLiftQuantizableSpotsAsFunctionsPass(quantization_specs));
      if (debugger_config.debugger_type() !=
          DebuggerConfig::DEBUGGER_TYPE_UNSPECIFIED) {
    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/cc/pass_pipeline.h

    // constant folding the transpose->convolution pattern. This is useful when
    // downstream pipeline (e.g. XLA) is more optimized when accepting NHWC formats.
    void AddProcessNchwTensorPasses(OpPassManager& pm);
    
    // Registers quantization pass pipelines. This is only required when running
    // MLIR opt binaries and not required when adding passes programmatically.
    void RegisterPassPipelines();
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 12:53:33 UTC 2024
    - 3.6K bytes
    - Viewed (0)
Back to top