Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

        // When begin/end values are dynamic, the ellipsis mask, if set, must refer
        // to the last dimension.
        int ellipsis_mask = op.getEllipsisMask();
        if (!(ellipsis_mask == 0 || ellipsis_mask == (1 << last_dim)))
          return rewriter.notifyMatchFailure(
              op,
              "requires that ellipsis_mask, if set, refer to the last dimension of "
              "input (when begin/end values are dynamic)");
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

        TF_Tensor:$value,
    
        DefaultValuedOptionalAttr<I64Attr, "0">:$begin_mask,
        DefaultValuedOptionalAttr<I64Attr, "0">:$end_mask,
        DefaultValuedOptionalAttr<I64Attr, "0">:$ellipsis_mask,
        DefaultValuedOptionalAttr<I64Attr, "0">:$new_axis_mask,
        DefaultValuedOptionalAttr<I64Attr, "0">:$shrink_axis_mask
      );
    
      let results = (outs);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top