Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for getDimensionNumbersAttr (0.16 sec)

  1. 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)
  2. 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)
Back to top