Home
last modified time | relevance | path

Searched refs:ExecuteKernelOnStream (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dstream_executor_util.h86 Status ExecuteKernelOnStream(const se::KernelBase& kernel,
H A Dkernel_thunk.cc118 return ExecuteKernelOnStream(*kernel, buffer_args, launch_dimensions, in ExecuteOnStream()
H A Dstream_executor_util.cc357 Status ExecuteKernelOnStream(const se::KernelBase& kernel, in ExecuteKernelOnStream() function
H A Djitrt_custom_calls.cc471 ExecuteKernelOnStream(*kernel, buffer_args, launch_dimensions, stream); in operator ()()