Home
last modified time | relevance | path

Searched refs:kernel_args (Results 1 – 9 of 9) 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
216 kernel_args->add_device_memory_argument(buf); in ExecuteKernelOnStream()
223 *kernel_args); in ExecuteKernelOnStream()
Dir_emitter_unnested.cc3385 std::unordered_map<const BufferAllocation*, llvm::Value*> kernel_args; in BuildKernelThunkFromBufferSlices() local
3390 kernel_args[*buffers_it] = arg_it; in BuildKernelThunkFromBufferSlices()
3426 loc = InBoundsGEP(kernel_args.at(buffer_slice.allocation()), in BuildKernelThunkFromBufferSlices()
3445 bindings_.SetTempBufferBase(kernel_args.at(*temp_buffer)); in BuildKernelThunkFromBufferSlices()
/external/mesa3d/src/gallium/drivers/radeonsi/
Dsi_compute.c657 uint32_t *kernel_args; in si_upload_compute_input() local
668 kernel_args = (uint32_t *)kernel_args_ptr; in si_upload_compute_input()
671 memcpy(kernel_args, info->input, program->input_size); in si_upload_compute_input()
674 COMPUTE_DBG(sctx->screen, "input %u : %u\n", i, kernel_args[i]); in si_upload_compute_input()
/external/mesa3d/src/gallium/drivers/llvmpipe/
Dlp_jit.h343 void *kernel_args; member
Dlp_jit.c414 LP_CHECK_MEMBER_OFFSET(struct lp_jit_cs_context, kernel_args, in lp_jit_create_cs_types()
Dlp_state_cs.c398 params.kernel_args = kernel_args_ptr; in generate_compute()
1249 csctx->cs.current.jit_context.kernel_args = input; in llvmpipe_cs_update_derived()
/external/mesa3d/src/gallium/auxiliary/gallivm/
Dlp_bld_tgsi.h285 LLVMValueRef kernel_args; member
Dlp_bld_nir_soa.c2093 bld.kernel_args_ptr = params->kernel_args; in lp_build_nir_soa()
/external/tensorflow/tensorflow/stream_executor/
Dstream.h2130 KernelArgsArray<sizeof...(args)> kernel_args; in ThenLaunch() local
2131 kernel.PackParams(&kernel_args, args...); in ThenLaunch()
2134 parent_->Launch(this, thread_dims, block_dims, kernel, kernel_args) in ThenLaunch()