Searched defs:AsyncRunTask (Results 1 – 1 of 1) sorted by relevance
325 struct AsyncRunTask { in ExecuteAsyncOnStreamImpl() struct326 CpuExecutable* executable; in ExecuteAsyncOnStreamImpl()327 ServiceExecutableRunOptions run_options; in ExecuteAsyncOnStreamImpl()328 std::vector<se::DeviceMemoryBase> unowning_buffers; in ExecuteAsyncOnStreamImpl()329 std::shared_ptr<std::vector<OwningDeviceMemory>> buffers; in ExecuteAsyncOnStreamImpl()330 HloExecutionProfile* hlo_execution_profile; in ExecuteAsyncOnStreamImpl()332 void operator()() { in ExecuteAsyncOnStreamImpl()