Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TILE (0.03 sec)

  1. tensorflow/compiler/mlir/tfrt/tests/tf_to_corert/attributes.mlir

      %multiples = "tf.Const"() { device = "/device:CPU:0", value = dense<[7,3]> : tensor<2xi32> } : () -> tensor<2xi32>
      // CHECK: T = !corert.string
      %output = "tf.Tile"(%arg, %multiples) { device = "/device:CPU:0" } : (tensor<1x2x!tf_type.string>, tensor<2xi32>) -> tensor<?x6x!tf_type.string>
      func.return %output : tensor<?x6x!tf_type.string>
    }
    
    // CHECK-LABEL: func @shape
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 00:18:59 UTC 2024
    - 4.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.h

    // A sharding with OTHER type may be REPLICATED if:
    // 'replicate_on_last_tile_dim' is true OR
    // 'last_tile_dims' is not empty
    // AND
    // other than replicated last tile dims, all other dims are not sharded.
    bool IsOtherReplicatedSharding(const xla::OpSharding& xla_sharding);
    
    // Returns whether the sharding is split sharding. i.e. A sharding with OTHER
    // type but not replicated.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Mar 28 22:18:34 UTC 2024
    - 6K bytes
    - Viewed (0)
Back to top