- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for num_tasks (0.16 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
mesh_shape_dim.value())); if (topology_proto.num_tasks() != num_tasks) return absl::InvalidArgumentError(absl::StrCat( "number of tasks from available TPU devices must be 'num_tasks' in '", kTopologyAttr, "' (", topology_proto.num_tasks(), "), got ", num_tasks)); if (topology_proto.num_tpu_devices_per_task() != num_tpus_per_task)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc
"(0, 0, 0, 0)"))); std::vector<std::string> MakeDeviceSet(int num_tasks, int num_devices_per_task) { std::vector<std::string> devices{ "/job:localhost/replica:0/task:0/device:CPU:0"}; devices.reserve(num_tasks * num_devices_per_task + num_tasks + 1); for (int task = 0; task < num_tasks; ++task) { devices.push_back(
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/tensorflow/utils/tpu_rewrite_device_util.h
// `num_replicas` = 4 // `num_cores_per_replica` = 2 // `topology_attr` (in proto debug string format) = // { // mesh_shape: 2 // mesh_shape: 2 // mesh_shape: 2 // num_tasks: 2 // num_tpu_devices_per_task: 4 // device_coordinates: 0 // device_coordinates: 0 // device_coordinates: 0 // device_coordinates: 0 // device_coordinates: 1
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/tests/tpu_rewrite.mlir
} } // ----- // The following topology is used in subsequent test cases: // Proto debug string: // mesh_shape: 1 // mesh_shape: 1 // mesh_shape: 1 // mesh_shape: 2 // num_tasks: 1 // num_tpu_devices_per_task: 2 // device_coordinates: 0 // device_coordinates: 0 // device_coordinates: 0 // device_coordinates: 0 // device_coordinates: 0 // device_coordinates: 0
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 172.9K bytes - Viewed (0)