Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for split_sizes (0.33 sec)

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

    // and they are not necessarily the same.
    //
    // For example, given the following IR:
    //
    // %split_sizes = "tf.Const"() {value = dense<[1, -1, 3]> : tensor<3xi32>}
    // %split_dim = "tf.Const"() {value = dense<1> : tensor<i32>}
    // %0:3 = "tf.SplitV"(%input, %split_sizes, %split_dim) :
    //                   (tensor<4x6xf32>, tensor<3xi32>, tensor<i32>) ->
    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/tf2xla/tests/legalize-tf.mlir

    func.func @splitv_dynamic(%input: tensor<?x6xf32>) -> (tensor<?x1xf32>, tensor<?x2xf32>, tensor<?x3xf32>) {
      %split_sizes = "tf.Const"() {value = dense<[1, 2, 3]> : tensor<3xi32>} : () -> tensor<3xi32>
      %split_dim = "tf.Const"() {value = dense<1> : tensor<i32>} : () -> tensor<i32>
      // CHECK: tf.SplitV
      %0:3 = "tf.SplitV"(%input, %split_sizes, %split_dim) : (tensor<?x6xf32>, tensor<3xi32>, tensor<i32>) -> (tensor<?x1xf32>, tensor<?x2xf32>, tensor<?x3xf32>)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
Back to top