- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for getDimensionNumbersAttr (0.28 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
<< "have same number of dimensions.\n"; } DenseElementsAttr feature_group_count_attr; xla::ConvolutionDimensionNumbers dnums; dnums.ParseFromString(op.getDimensionNumbersAttr().getValue().str()); if (dnums.input_spatial_dimensions_size() != dnums.kernel_spatial_dimensions_size()) { return op.emitOpError() << "Both arguments to convolution must have "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc
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( "dimension_numbers", xla::ConvertConvDimensionNumbers(dnums, &rewriter));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)