- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for getBatchGroupCountAttr (0.34 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/nchw_convolution_to_nhwc.cc
/*window_reversal=*/op.getWindowReversalAttr(), /*dimension_numbers=*/new_dimension_nums, /*feature_group_count=*/op.getFeatureGroupCountAttr(), /*batch_group_count=*/op.getBatchGroupCountAttr(), /*precision_config=*/op.getPrecisionConfigAttr()); // Transpose the output of the `ConvolutionOp` back to the original op's // output shape so that users' shapes match.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
"feature_group_count", rewriter.getI64IntegerAttr(feature_group_count_val)); auto batch_group_count_named_attr = rewriter.getNamedAttr("batch_group_count", op.getBatchGroupCountAttr()); xla::ConvolutionDimensionNumbers dnums; dnums.ParseFromString(op.getDimensionNumbersAttr().getValue().str()); auto dimension_numbers_named_attr = rewriter.getNamedAttr(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)