Searched refs:function_key (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | process_function_library_runtime.h | 129 const string& function_key, const string& device_name,
133 FunctionLibraryRuntime::Handle GetHandle(const string& function_key) const;
265 const string& function_key, int num_outputs, in MultiDeviceFunctionData()
269 function_key_(function_key), in MultiDeviceFunctionData()
329 const string& function_key, const string& device_name,
378 const string& function_key);
407 const string& function_key) in FunctionData() argument
410 function_key_(function_key) {} in FunctionData()
413 const string& function_key() { return function_key_; } in function_key() function
|
| D | process_function_library_runtime.cc | 252 const string& function_key, const string& device_name, in AddHandle() argument
255 return AddHandleLocked(function_key, device_name, local_handle); in AddHandle()
259 const string& function_key, const string& device_name, in AddHandleLocked() argument
263 absl::make_unique<FunctionData>(device_name, local_handle, function_key); in AddHandleLocked()
264 table_[function_key] = h; in AddHandleLocked()
271 std::unique_ptr<MultiDeviceFunctionData> data, const string& function_key) { in AddMultiDeviceHandle() argument
275 table_[function_key] = h; in AddMultiDeviceHandle()
281 const string& function_key) const { in GetHandle()
283 return gtl::FindWithDefault(table_, function_key, kInvalidHandle); in GetHandle()
673 const string& function_key = Canonicalize(function_name, attrs, options); in InstantiateMultiDevice() local
[all …]
|
| /external/tensorflow/tensorflow/python/profiler/ |
| D | pprof_profiler.py | 129 function_key = (file_path, function_name, function_start_line)
130 if function_key in self._function_key_to_function:
131 return self._function_key_to_function[function_key].id
140 self._function_key_to_function[function_key] = function
|