Searched refs:function_key (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 130 const string& function_key, const string& device_name,
134 FunctionLibraryRuntime::Handle GetHandle(const string& function_key) const;
300 const string& function_key, int num_outputs, in MultiDeviceFunctionData()
304 function_key_(function_key), in MultiDeviceFunctionData()
350 const string& function_key, const string& device_name,
399 const string& function_key);
463 const string& function_key) in FunctionData() argument
466 function_key_(function_key) {} in FunctionData()
469 const string& function_key() { return function_key_; } in function_key() function
|
| D | process_function_library_runtime.cc | 271 const string& function_key, const string& device_name, in AddHandle() argument
274 return AddHandleLocked(function_key, device_name, local_handle); in AddHandle()
278 const string& function_key, const string& device_name, in AddHandleLocked() argument
282 std::make_unique<FunctionData>(device_name, local_handle, function_key); in AddHandleLocked()
283 table_[function_key] = h; in AddHandleLocked()
290 std::unique_ptr<MultiDeviceFunctionData> data, const string& function_key) { in AddMultiDeviceHandle() argument
294 table_[function_key] = h; in AddMultiDeviceHandle()
310 const string& function_key) const { in GetHandle()
312 return gtl::FindWithDefault(table_, function_key, kInvalidHandle); in GetHandle()
760 const string& function_key = Canonicalize(function_name, attrs, options); in InstantiateMultiDevice() local
[all …]
|
| /external/tensorflow/tensorflow/python/profiler/ |
| D | pprof_profiler.py | 125 function_key = (file_path, function_name, function_start_line)
126 if function_key in self._function_key_to_function:
127 return self._function_key_to_function[function_key].id
136 self._function_key_to_function[function_key] = function
|