- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for XlaSpmdFullToShardShape (0.32 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/extract_outside_compilation.mlir
// CHECK: }, { // CHECK: "tf_device.cluster" // CHECK: %[[A:.+]] = "tf.OpA" // CHECK: %[[A_SHARD:.+]] = "tf.XlaSpmdFullToShardShape"(%[[A]]) <{dim = -1 : i64, manual_sharding = "\08\03\1A\02\02\01\22\02\00\01", unspecified_dims = []}> : (tensor<2x2xi64>) -> tensor<1x2xi64> // CHECK: %[[B:.+]] = "tf._XlaHostComputeMlir"(%[[A_SHARD]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 129.6K bytes - Viewed (0)