Searched refs:MultiDeviceFunctionData (Results 1 – 2 of 2) sorted by relevance
298 struct MultiDeviceFunctionData { struct299 MultiDeviceFunctionData(const string& function_name, in MultiDeviceFunctionData() argument343 MultiDeviceFunctionData* IsMultiDevice( argument398 const std::unique_ptr<MultiDeviceFunctionData> data,435 const MultiDeviceFunctionData* data) const;439 const MultiDeviceFunctionData** data) const;529 std::unique_ptr<MultiDeviceFunctionData>>
290 std::unique_ptr<MultiDeviceFunctionData> data, const string& function_key) { in AddMultiDeviceHandle()325 const MultiDeviceFunctionData& data = *miter->second; in GetHandleOnDevice()358 ProcessFunctionLibraryRuntime::MultiDeviceFunctionData*848 auto data = std::make_unique<MultiDeviceFunctionData>( in InstantiateMultiDevice()1162 MultiDeviceFunctionData* data = IsMultiDevice(handle); in GetOutputDevices()1209 const MultiDeviceFunctionData** data) const { in PrepareRunMultiDevice()1239 const MultiDeviceFunctionData* data) const { in GetOrderedSubgraphs()1266 const MultiDeviceFunctionData* data; in RunMultiDeviceSync()1363 const MultiDeviceFunctionData* data; in RunMultiDeviceAsync()1552 std::unique_ptr<MultiDeviceFunctionData> mdata; in ReleaseMultiDeviceHandle()[all …]