- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for MatchOutputDimensionNumbers (0.24 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc
return success(MatchInputDimensionNumbers(dimension_nums) && MatchKernelDimensionNumbers(dimension_nums) && MatchOutputDimensionNumbers(dimension_nums)); } void rewrite(mlir::stablehlo::ConvolutionOp op, PatternRewriter& rewriter) const override { // Transpose the input tensor: [b, f, 0, 1] => [b, 0, 1, f]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.2K bytes - Viewed (0)