- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for perf (0.12 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
TF_Status* status) const { absl::optional<std::vector<std::unique_ptr<ParallelTensor>>> result; // Compute per-device per-output tensors std::vector<std::vector<TensorHandlePtr>> per_device_output_tensors; per_device_output_tensors.reserve(underlying_devices_.size()); int first_op_output_count = 0; StatusPtr first_bad_status(nullptr);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device.cc
// or consider other ways of packing/unpacking parallel tensors. if (operation_name == std::string("TPUReplicatedInput")) { // Special-cased operation for packing per-device tensors into one parallel // tensor. if (inputs.size() != parallel_device.num_underlying_devices()) { std::string message(absl::StrCat( "The parallel device ", parallel_device_name, " expected ",
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
const char task1_name[] = "/job:localhost/replica:0/task:1/device:CPU:0"; const char task2_name[] = "/job:localhost/replica:0/task:2/device:CPU:0"; // Create one variable per task. TFE_TensorHandle* h0 = TestVariable(ctx, 1.0, task1_name); TFE_TensorHandle* h1 = TestVariable(ctx, 2.0, task2_name); TFE_TensorHandle* h2 = TestVariable(ctx, 3.0, task0_name);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0)