Searched refs:per_device_output_tensors (Results 1 – 1 of 1) sorted by relevance
345 std::vector<std::vector<TensorHandlePtr>> per_device_output_tensors; in Join() local346 per_device_output_tensors.reserve(underlying_devices_.size()); in Join()352 per_device_output_tensors.push_back(device_thread->Join(status)); in Join()367 first_op_output_count = per_device_output_tensors.rbegin()->size(); in Join()370 per_device_output_tensors.rbegin()->size() != first_op_output_count) { in Join()390 components.push_back(std::move(per_device_output_tensors[j][i])); in Join()