Home
last modified time | relevance | path

Searched defs:kernel_args (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dstream_executor_util.cc214 auto kernel_args = absl::make_unique<se::KernelArgsArray<kKernelArgsLimit>>(); in ExecuteKernelOnStream() local
Dir_emitter_unnested.cc3385 std::unordered_map<const BufferAllocation*, llvm::Value*> kernel_args; in BuildKernelThunkFromBufferSlices() local
/external/tensorflow/tensorflow/stream_executor/
Dstream.h2130 KernelArgsArray<sizeof...(args)> kernel_args; in ThenLaunch() local
/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_compute.c657 uint32_t *kernel_args; in si_upload_compute_input() local
/external/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_jit.h343 void *kernel_args; member
/external/mesa3d/src/gallium/auxiliary/gallivm/
Dlp_bld_tgsi.h285 LLVMValueRef kernel_args; member