Home
last modified time | relevance | path

Searched defs:add_kernel (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_triton_kernels.py1085 def add_kernel( function
1418 def add_kernel( function
1457 def add_kernel( function
1513 def add_kernel( function
2430 def add_kernel( function
2473 def add_kernel( function
2528 def add_kernel( function
/aosp_15_r20/external/pytorch/test/cpp_extensions/
H A Dcuda_dlink_extension_kernel.cu9 __global__ void add_kernel(const float* a, const float* b, float* output, int size) { in add_kernel() function
/aosp_15_r20/external/pytorch/test/mobile/nnc/
H A Dtest_nnc_backend.cpp53 int add_kernel(void** args) { in add_kernel() function
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dtriton_utils.py18 def add_kernel( function
/aosp_15_r20/external/pytorch/torch/_higher_order_ops/
H A Dtriton_kernel_wrap.py44 def add_kernel(self, kernel) -> int: member in KernelSideTable
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/framework/
Dkernel_def.pb.h1039 inline ::tensorflow::KernelDef* KernelList::add_kernel() { in add_kernel() function