- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for replica_device_ids (0.31 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.h
// computation_count: 2 // computation_devices { // replica_device_ids: 0 // replica_device_ids: 4 // replica_device_ids: 2 // replica_device_ids: 6 // } // computation_devices { // replica_device_ids: 1 // replica_device_ids: 5 // replica_device_ids: 3 // replica_device_ids: 7 // } // }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 11.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc
EXPECT_EQ(computation_device_0.replica_device_ids(0), 0); EXPECT_EQ(computation_device_0.replica_device_ids(1), 4); EXPECT_EQ(computation_device_0.replica_device_ids(2), 2); EXPECT_EQ(computation_device_0.replica_device_ids(3), 6); EXPECT_EQ(computation_device_1.replica_device_ids(0), 1); EXPECT_EQ(computation_device_1.replica_device_ids(1), 5); EXPECT_EQ(computation_device_1.replica_device_ids(2), 3);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 46.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/ifrt/rewrite_cluster_to_ifrt_call.mlir
// CHECK-SAME: tfrt_ifrt_serving.program_id = [[PROGRAM_ID]] : i64 // CHECK: return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Feb 17 07:28:40 UTC 2024 - 9K bytes - Viewed (0)