Searched refs:MultiDeviceFunctionData (Results 1 – 2 of 2) sorted by relevance
263 struct MultiDeviceFunctionData { struct264 MultiDeviceFunctionData(const string& function_name, in MultiDeviceFunctionData() argument305 MultiDeviceFunctionData* IsMultiDevice( argument377 const std::unique_ptr<MultiDeviceFunctionData> data,473 std::unique_ptr<MultiDeviceFunctionData>>
271 std::unique_ptr<MultiDeviceFunctionData> data, const string& function_key) { in AddMultiDeviceHandle()301 const MultiDeviceFunctionData& data = *miter->second; in GetHandleOnDevice()334 ProcessFunctionLibraryRuntime::MultiDeviceFunctionData*750 auto data = absl::make_unique<MultiDeviceFunctionData>( in InstantiateMultiDevice()1019 MultiDeviceFunctionData* data = IsMultiDevice(handle); in GetOutputDevices()1083 const MultiDeviceFunctionData* data = IsMultiDevice(handle); in RunMultiDevice()1288 std::unique_ptr<MultiDeviceFunctionData> mdata; in ReleaseMultiDeviceHandle()1646 const MultiDeviceFunctionData* data = IsMultiDevice(handle); in Run()