- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for XlaSpmdShardToFullShape (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/extract_outside_compilation.mlir
// CHECK-SAME: recv_key = "host_compute_channel_0_retvals" // CHECK-SAME: send_key = "host_compute_channel_0_args" // CHECK: %[[B_FULL:.+]] = "tf.XlaSpmdShardToFullShape"(%[[B]]) <{dim = -1 : i64, full_shape = #tf_type.shape<2x2>, manual_sharding = "\08\03\1A\02\02\01\22\02\00\01", unspecified_dims = []}> : (tensor<1x2xi64>) -> tensor<2x2xi64> // CHECK: "tf.OpC"(%[[B_FULL]])
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 31 08:59:10 UTC 2023 - 129.6K bytes - Viewed (0)