- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 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/c/eager/c_api_test_util.cc
<< ", task: " << i; } return server_def; } tensorflow::ServerDef GetServerDef(int num_tasks) { return GetServerDef("localhost", num_tasks); } tensorflow::ServerDef GetMultiClientServerDef(const std::string& job_name, int num_tasks, int num_virtual_gpus) { tensorflow::ServerDef server_def;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 22:37:46 UTC 2024 - 23.5K 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) -
pkg/kubelet/cm/topologymanager/bitmask/bitmask_test.go
// Iterate all masks and count them. numMasks := 0 IterateBitMasks(bits, func(BitMask) { numMasks++ }) // Compare the number of masks generated to the expected amount. if expectedNumMasks != numMasks { t.Errorf("Expected to iterate %v masks, got %v", expectedNumMasks, numMasks) } } } func TestIsLessThan(t *testing.T) { tcases := []struct {
Registered: Sat Jun 15 01:39:40 UTC 2024 - Last Modified: Thu Nov 03 09:45:09 UTC 2022 - 16.2K bytes - Viewed (0)