- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for core_to_host (0.11 sec)
-
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util_test.cc
[&](mlir::tf_device::ClusterOp descendant) { cluster = descendant; }); llvm::SmallVector<std::string, 8> core_to_host; EXPECT_TRUE(mlir::succeeded(GetDeviceToHostMap(cluster, core_to_host))); EXPECT_EQ(core_to_host.size(), 2); EXPECT_EQ(core_to_host[0], "/job:localhost/replica:0/task:0/device:CPU:0"); EXPECT_EQ(core_to_host[1], "/job:localhost/replica:0/task:0/device:CPU:0"); }
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.cc
int num_cores_per_replica = GetNumCoresPerReplica(cluster); llvm::SmallVector<std::string, 8> core_to_host; core_to_host.reserve(num_cores_per_replica); for (int core = 0; core < num_cores_per_replica; ++core) { core_to_host.push_back(GetDeviceAliasForHostOfLogicalCore(core)); } return core_to_host; } // Get the map from `core` to host device for a non-replicated TPU cluster.
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/tf2xla/internal/passes/extract_outside_compilation.cc
SmallVector<std::string, 8> core_to_host; if (failed(tensorflow::GetDeviceToHostMap(device_cluster, core_to_host))) return signalPassFailure(); if (failed(CreateParallelExecuteForOutsideCompilation( device_cluster, tmp_parallel_execute_ops, is_map_oc, core_to_host, tensorflow::HasTPUDevice(devices)))) return signalPassFailure();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.h
// is the associated virtual device name (`TPU_REPLICATED_HOST_i`). mlir::LogicalResult GetDeviceToHostMap( mlir::tf_device::ClusterOp cluster, llvm::SmallVector<std::string, 8>& core_to_host); // Returns the first TPU device, for use in the non-replicated case. The list of // TPU devices is retrived from `op`'s module ancestor. mlir::LogicalResult GetNonReplicatedTPU0(mlir::Operation* op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 11.3K bytes - Viewed (0)