Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for nchw_tensor_format_processing_pipeline (0.27 sec)

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

      // NCHW->NHWC convolution conversion).
      pm.addNestedPass<func::FuncOp>(createFoldConstantTransposePass());
    }
    
    void RegisterPassPipelines() {
      static PassPipelineRegistration<> nchw_tensor_format_processing_pipeline(
          /*arg=*/"stablehlo-process-nchw-tensor",
          /*description=*/"Optimizes tensors with NCHW format.",
          AddProcessNchwTensorPasses);
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 10 04:07:09 UTC 2024
    - 8.1K bytes
    - Viewed (0)
Back to top