Searched defs:CustomAddKernel (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/examples/runtime_extend/src/ |
| D | custom_add_kernel.cc | 30 class CustomAddKernel : public Kernel { class
32 CustomAddKernel(const std::vector<MSTensor> &inputs, const std::vector<MSTensor> &outputs, in CustomAddKernel() function in mindspore::kernel::CustomAddKernel
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/graph_kernel/custom/aot_test_files/ |
| D | add.cu | 29 __global__ void CustomAddKernel(float *input1, float *input2, float *output, size_t size) { in CustomAddKernel() function
|
| D | add_with_attr.cu | 21 __global__ void CustomAddKernel(float *input1, float *input2, float *output, float *tmp, float scal… in CustomAddKernel() function
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/registry/ |
| D | registry_gpu_custom_op_test.cc | 153 class CustomAddKernel : public kernel::Kernel { class
155 CustomAddKernel(const std::vector<MSTensor> &inputs, const std::vector<MSTensor> &outputs, in CustomAddKernel() function in mindspore::CustomAddKernel
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/ |
| D | kernel_executor_tests.cc | 53 class CustomAddKernel : public kernel::Kernel { class
55 CustomAddKernel(const std::vector<MSTensor> &inputs, const std::vector<MSTensor> &outputs, in CustomAddKernel() function in mindspore::__anon4626911c0111::CustomAddKernel
|