Searched refs:kernel_args (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | stream_executor_util.cc | 214 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()
|
D | ir_emitter_unnested.cc | 3385 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/ |
D | si_compute.c | 657 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/ |
D | lp_jit.h | 343 void *kernel_args; member
|
D | lp_jit.c | 414 LP_CHECK_MEMBER_OFFSET(struct lp_jit_cs_context, kernel_args, in lp_jit_create_cs_types()
|
D | lp_state_cs.c | 398 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/ |
D | lp_bld_tgsi.h | 285 LLVMValueRef kernel_args; member
|
D | lp_bld_nir_soa.c | 2093 bld.kernel_args_ptr = params->kernel_args; in lp_build_nir_soa()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | stream.h | 2130 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()
|