| /external/pytorch/aten/src/ATen/cuda/ |
| D | jiterator.cu | 78 constexpr int kernel_args = 3; in launch_jitted_vectorized_kernel_dynamic() local
99 constexpr int kernel_args = 7; in launch_jitted_vectorized_kernel_dynamic() local
165 constexpr int kernel_args = 7; in launch_jitted_unrolled_kernel_dynamic() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | stream_executor_util.cc | 350 auto kernel_args = std::make_unique<se::KernelArgsArray<n>>(); in MakeKernelArgs() local
361 std::unique_ptr<se::KernelArgsArrayBase> kernel_args; in ExecuteKernelOnStream() local
|
| D | ir_emitter_unnested.cc | 3000 absl::flat_hash_map<const BufferAllocation*, llvm::Value*> kernel_args; in BuildKernelThunkImpl() local
|
| /external/executorch/runtime/executor/test/ |
| D | executor_test.cpp | 105 EValue* kernel_args[5]; in TEST_F() local
|
| /external/cronet/stable/third_party/llvm-libc/src/utils/gpu/loader/nvptx/ |
| D | nvptx-loader.cpp | 172 args_t kernel_args, bool print_resource_usage) { in launch_kernel()
|
| /external/cronet/tot/third_party/llvm-libc/src/utils/gpu/loader/nvptx/ |
| D | nvptx-loader.cpp | 172 args_t kernel_args, bool print_resource_usage) { in launch_kernel()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| D | kernel_ops_pattern.cc | 411 SmallVector<Value, 4> kernel_args; in Rewrite() local
|
| /external/mesa3d/src/gallium/drivers/llvmpipe/ |
| D | lp_jit.h | 326 const void *kernel_args; member
|
| /external/cronet/tot/third_party/llvm-libc/src/utils/gpu/loader/amdgpu/ |
| D | amdhsa-loader.cpp | 167 const char *kernel_name, args_t kernel_args, in launch_kernel()
|
| /external/cronet/stable/third_party/llvm-libc/src/utils/gpu/loader/amdgpu/ |
| D | amdhsa-loader.cpp | 167 const char *kernel_name, args_t kernel_args, in launch_kernel()
|
| /external/mesa3d/src/gallium/frontends/rusticl/core/ |
| D | program.rs | 111 let kernel_args: HashSet<_> = self in build_nirs() localVariable
|
| /external/mesa3d/src/gallium/drivers/radeonsi/ |
| D | si_compute.c | 734 uint32_t *kernel_args; in si_upload_compute_input() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream.h | 1635 KernelArgsArray<sizeof...(args)> kernel_args; in ThenLaunch() local
|
| /external/mesa3d/src/gallium/auxiliary/gallivm/ |
| D | lp_bld_tgsi.h | 289 LLVMValueRef kernel_args; member
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | Reduce.cuh | 930 constexpr int kernel_args = 1; in launch_jitted_reduce_kernel() local
|
| /external/pytorch/torch/_inductor/codegen/ |
| D | cpp.py | 4544 def define_kernel(self, src_code, nodes, kernel_args=None): argument
|
| /external/pytorch/torch/_inductor/ |
| D | ir.py | 4713 def __init__(self, *, kernel_idx, grid, kernel_args): argument
|
| /external/mesa3d/src/intel/vulkan/ |
| D | anv_private.h | 4651 struct brw_kernel_arg_desc * kernel_args; member
|