Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for getSizeSplits (0.19 sec)

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

          return failure();
    
        // We can only match when the split sizes is a constant int vector.
        DenseIntElementsAttr split_sizes_attr;
        if (!matchPattern(op.getSizeSplits(), m_Constant(&split_sizes_attr)))
          return failure();
    
        // Get each chunck's size along the dimension to split. It may contain
        // dynamic sizes and we need to update it if so.
    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