Searched refs:function_data (Results 1 – 5 of 5) sorted by relevance
149 FunctionData* function_data = nullptr; in Run() local153 function_data = &function_data_[handle]; in Run()156 EagerClient* eager_client = function_data->eager_client.get(); in Run()162 EagerOperation* op = function_data->op.get(); in Run()173 if (function_data->ret_indices.has_value()) { in Run()174 for (const int ret_index : function_data->ret_indices.value()) { in Run()202 remote_op->set_device(function_data->target); in Run()258 FunctionData* function_data = nullptr; in CleanUp() local262 function_data = &function_data_[handle]; in CleanUp()265 EagerClient* eager_client = function_data->eager_client.get(); in CleanUp()
175 for (auto& function_data : function_data_) { in ~ClusterFunctionLibraryRuntime() local176 worker_session_->worker_cache()->ReleaseWorker(function_data.target, in ~ClusterFunctionLibraryRuntime()177 function_data.wi); in ~ClusterFunctionLibraryRuntime()261 FunctionData* function_data = nullptr; in Run() local265 function_data = &function_data_[handle]; in Run()268 WorkerInterface* wi = function_data->wi; in Run()278 req->set_graph_handle(function_data->graph_handle); in Run()281 for (const auto& send_key : function_data->send_keys) { in Run()287 const std::vector<string>& recv_keys = function_data->recv_keys; in Run()365 FunctionData* function_data = nullptr; in CleanUp() local[all …]
256 const CFIUnwindDataRow* function_data = in FindCFIRowForPC() local265 memcpy(&row, function_data + i, sizeof(CFIUnwindDataRow)); in FindCFIRowForPC()
318 FunctionData* function_data = iter->second.get(); in GetHandleOnDevice() local319 if (function_data->target_device() != device_name) { in GetHandleOnDevice()322 return function_data->local_handle(); in GetHandleOnDevice()330 FunctionData* function_data = iter->second.get(); in GetDeviceName() local331 return function_data->target_device(); in GetDeviceName()1489 FunctionData* function_data = iter->second.get(); in RunInternal() local1490 target_device = function_data->target_device(); in RunInternal()1491 local_handle = function_data->local_handle(); in RunInternal()
106 } function_data; typedef108 extern const std::unordered_map<std::string, function_data> name_to_funcptr_map;9265 const std::unordered_map<std::string, function_data> name_to_funcptr_map = {