Home
last modified time | relevance | path

Searched defs:CustomAddKernel (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/lite/examples/runtime_extend/src/
Dcustom_add_kernel.cc31 class CustomAddKernel : public Kernel { class
33 CustomAddKernel(const std::vector<MSTensor> &inputs, const std::vector<MSTensor> &outputs, in CustomAddKernel() function in mindspore::kernel::CustomAddKernel
/third_party/mindspore/mindspore/lite/test/ut/src/registry/
Dregistry_gpu_custom_op_test.cc156 class CustomAddKernel : public kernel::Kernel { class
158 CustomAddKernel(const std::vector<MSTensor> &inputs, const std::vector<MSTensor> &outputs, in CustomAddKernel() function in mindspore::CustomAddKernel