Searched refs:comp_rets (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.cc | 1342 std::vector<FunctionRet> comp_rets; in RunMultiDeviceSync() local
1343 RunInternal(opts_copy, comp_handle, comp_args.args, &comp_rets, in RunMultiDeviceSync()
1403 std::vector<FunctionRet>* comp_rets = new std::vector<FunctionRet>; in RunMultiDeviceAsync() local
1406 auto component_fn_callback = [comp_rets, rets, comp_data, refcounted_done, in RunMultiDeviceAsync()
1424 for (int i = 0; i < comp_rets->size(); ++i) { in RunMultiDeviceAsync()
1425 (*rets)[comp_data.ret_indices[i]] = (*comp_rets)[i]; in RunMultiDeviceAsync()
1428 delete comp_rets; in RunMultiDeviceAsync()
1449 TensorsToFunctionRetsDoneCallback(comp_rets, comp_tensor_rets, in RunMultiDeviceAsync()
1458 RunInternal(opts_copy, comp_handle, comp_args.args, comp_rets, in RunMultiDeviceAsync()
|