| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A 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
|
| H A D | ir_emitter_unnested.cc | 3000 absl::flat_hash_map<const BufferAllocation*, llvm::Value*> kernel_args; in BuildKernelThunkImpl() local
|
| /aosp_15_r20/device/google/cuttlefish/host/commands/assemble_cvd/ |
| D | boot_config.cc | 93 const std::string& kernel_args, in WriteEnvironment()
|
| /aosp_15_r20/external/executorch/runtime/executor/test/ |
| H A D | executor_test.cpp | 105 EValue* kernel_args[5]; in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| H A D | kernel_ops_pattern.cc | 411 SmallVector<Value, 4> kernel_args; in Rewrite() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/llvmpipe/ |
| H A D | lp_jit.h | 326 const void *kernel_args; member
|
| /aosp_15_r20/external/mesa3d/src/gallium/frontends/rusticl/core/ |
| H A D | program.rs | 111 let kernel_args: HashSet<_> = self in build_nirs() localVariable
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/radeonsi/ |
| H A D | si_compute.c | 733 uint32_t *kernel_args; in si_upload_compute_input() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | stream.h | 1635 KernelArgsArray<sizeof...(args)> kernel_args; in ThenLaunch() local
|
| /aosp_15_r20/external/mesa3d/src/gallium/auxiliary/gallivm/ |
| H A D | lp_bld_tgsi.h | 289 LLVMValueRef kernel_args; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Reduce.cuh | 930 constexpr int kernel_args = 1; in launch_jitted_reduce_kernel() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp.py | 4544 def define_kernel(self, src_code, nodes, kernel_args=None): argument
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | ir.py | 4713 def __init__(self, *, kernel_idx, grid, kernel_args): argument
|
| /aosp_15_r20/external/mesa3d/src/intel/vulkan/ |
| H A D | anv_private.h | 4503 struct brw_kernel_arg_desc * kernel_args; member
|