Searched refs:comp_tensor_rets (Results 1 – 1 of 1) sorted by relevance
1317 std::vector<Tensor> comp_tensor_rets; in RunMultiDeviceSync() local1320 &comp_tensor_rets); in RunMultiDeviceSync()1330 for (int i = 0; i < comp_tensor_rets.size(); ++i) { in RunMultiDeviceSync()1331 (*rets)[comp_data.ret_indices[i]] = comp_tensor_rets[i]; in RunMultiDeviceSync()1445 std::vector<Tensor>* comp_tensor_rets = new std::vector<Tensor>; in RunMultiDeviceAsync() local1448 comp_tensor_rets, in RunMultiDeviceAsync()1449 TensorsToFunctionRetsDoneCallback(comp_rets, comp_tensor_rets, in RunMultiDeviceAsync()