Home
last modified time | relevance | path

Searched refs:MultiDeviceFunctionData (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.h180 struct MultiDeviceFunctionData { struct
181 MultiDeviceFunctionData(const string& function_name, in MultiDeviceFunctionData() function
226 MultiDeviceFunctionData* IsMultiDevice(
235 const std::unique_ptr<MultiDeviceFunctionData> data,
306 std::unique_ptr<MultiDeviceFunctionData>>
Dprocess_function_library_runtime.cc218 std::unique_ptr<MultiDeviceFunctionData> data, const string& function_key) { in AddMultiDeviceHandle()
268 ProcessFunctionLibraryRuntime::MultiDeviceFunctionData*
576 std::unique_ptr<MultiDeviceFunctionData> data = in InstantiateMultiDevice()
577 MakeUnique<MultiDeviceFunctionData>(function_name, function_key, in InstantiateMultiDevice()
704 const MultiDeviceFunctionData* data = IsMultiDevice(handle); in GetOutputDevices()
752 const MultiDeviceFunctionData* data = IsMultiDevice(handle); in RunMultiDevice()
855 std::unique_ptr<MultiDeviceFunctionData> mdata; in ReleaseMultiDeviceHandle()