Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for getBatchGroupCountAttr (0.26 sec)

  2. github.com/tensorflow/tensorflow

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