Searched refs:at_tensors (Results 1 – 1 of 1) sorted by relevance
589 std::vector<at::Tensor> at_tensors; in CollectSyncTensors() local616 at_tensors.push_back(*tensor_data); in CollectSyncTensors()622 if (!at_tensors.empty()) { in CollectSyncTensors()623 TORCH_LAZY_COUNTER("SyncTensorsToData", at_tensors.size()); in CollectSyncTensors()625 CreateTensorsData(at_tensors, devices); in CollectSyncTensors()