Home
last modified time | relevance | path

Searched defs:kernel_ (Results 1 – 25 of 38) sorted by relevance

12

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/opencl/kernel/
Dbatchnorm.h49 cl::Kernel kernel_; variable
Dlayer_norm.h53 cl::Kernel kernel_; variable
Dstack.h40 cl::Kernel kernel_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dmatmul_cpu_kernel_func.h51 KernelBase *kernel_ = nullptr; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/mkldnn/
Dpooling_cpu_kernel.h59 std::vector<int64_t> kernel_; variable
Dpooling_grad_cpu_kernel.h103 std::vector<int64_t> kernel_; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/cxx_api/kernel_executor/
Dkernel_executor_impl.h55 kernel::KernelExec *kernel_ = nullptr; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/kernel/cpu/nnacl/
Dnnacl_kernel.h52 KernelBase *kernel_ = nullptr; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Dcustom_actor.h52 AnfNodeWeakPtr kernel_; variable
Dkernel_actor.h168 CNodePtr kernel_; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/
Dexecution_plan.cc79 for (auto kernel_ : kernel_list) { in CalcTensorRefCount() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/string/
Dnormalize.cc47 kernel::LiteKernel *kernel_ = nullptr; member in mindspore::TestNormalize
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/
Dlite_mindrt.h96 kernel::KernelExec *kernel_; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/common/draw/adapter_graphs/
Dsub_graph_kernel_adapter_graph.h40 : kernel_(kernel), filter_(std::move(mark_filter)) {} in kernel_() function
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/
Dir_converter.h88 const CNodePtr kernel_; member
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/fp32/
Dnon_max_suppression_fp32_tests.cc50 kernel::LiteKernel *kernel_ = nullptr; member in mindspore::TestNMSFp32
Done_hot_fp32_test.cc47 kernel::LiteKernel *kernel_ = nullptr; member in mindspore::TestOneHotFp32
Dl2norm_fp32_test.cc40 kernel::LiteKernel *kernel_ = nullptr; member in mindspore::TestL2NormFp32
Dscale_fp32_tests.cc54 kernel::LiteKernel *kernel_ = nullptr; member in mindspore::TestScaleFp32
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/kash/
Dkernel_pack.h120 FlexArray *kernel_; variable
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/dvm/
Ddvm_kernel_mod.h91 dvm::Kernel kernel_; variable
/third_party/mindspore/mindspore-src/source/mindspore/lite/test/ut/src/runtime/kernel/arm/int8/
Dresize_bilinear_int8_tests.cc44 kernel::LiteKernel *kernel_ = nullptr; member in mindspore::TestResizeBilinearInt8
Dscale_int8.cc49 kernel::LiteKernel *kernel_ = nullptr; member in mindspore::TestScaleInt8
Dresize_nearest_neighbor_int8_tests.cc45 kernel::LiteKernel *kernel_ = nullptr; member in mindspore::TestResizeNearestNeighborInt8
/third_party/mindspore/mindspore-src/source/mindspore/lite/examples/runtime_gpu_extend/src/
Dcustom_add_kernel_gpu.cc187 cl::Kernel kernel_; member in mindspore::custom_gpu_demo::CustomAddKernelGpu

12