Searched defs:kernel_params (Results 1 – 11 of 11) sorted by relevance
| /aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/ |
| H A D | Pool.cpp | 87 Kernel2dParams kernel_params = create_kernel2d_params( in add_max_pool2d_node() local
162 Kernel2dParams kernel_params = in add_avg_pool2d_node() local
|
| H A D | Convolution.cpp | 351 Kernel2dParams kernel_params = create_kernel2d_params( in add_conv2d_node() local
452 Kernel1dParams kernel_params = { in add_conv1d_node() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/ |
| H A D | sdp_utils_cpp.cpp | 60 SDPBackend select_sdp_backend_cpp(sdp_params const& kernel_params) { in select_sdp_backend_cpp()
|
| H A D | attention.cpp | 428 sdp::sdp_params kernel_params{query_, key, value, attn_mask_, dropout_p, is_causal}; in _fused_sdp_choice_cpp() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| H A D | tf_kernel_to_llvm_pass.cc | 202 auto kernel_params = generateParamsArray(launch_op, adaptor, rewriter); in matchAndRewrite() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/ |
| H A D | attention.cu | 563 sdp::sdp_params kernel_params{q, k, v, mask, 0.0, false}; in native_multi_head_attention_cuda() local
831 sdp::sdp_params kernel_params{query_, key, value, attn_mask_, dropout_p, is_causal}; in _fused_sdp_choice_cuda() local
|
| H A D | sdp_utils.cpp | 673 SDPBackend select_sdp_backend(sdp_params const& kernel_params) { in select_sdp_backend()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_gpu_executor.cc | 469 void** kernel_params = const_cast<void**>(args.argument_addresses().data()); in Launch() local
|
| H A D | cuda_driver.cc | 464 void** kernel_params, void** extra) { in LaunchKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_driver.cc | 409 GpuStreamHandle stream, void** kernel_params, void** extra) { in LaunchKernel()
|
| /aosp_15_r20/external/mesa3d/src/freedreno/ir3/ |
| H A D | ir3_shader.h | 201 unsigned kernel_params; member
|