- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for replica_device_ids (0.29 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)