Searched defs:InitKernel (Results 1 – 4 of 4) sorted by relevance
151 int NNACLKernel::InitKernel(const TypeId &data_type, const lite::InnerContext *ctx) { in InitKernel() function in mindspore::nnacl::NNACLKernel
38 bool CustomBiShengKernel::InitKernel(const AnfNodePtr &kernel_node) { in InitKernel() function in mindspore::kernel::CustomBiShengKernel
35 bool KernelPacketInitializer::InitKernel(const CNodePtr &real_node, const KernelModPtr &real_kernel… in InitKernel() function in mindspore::kernel::KernelPacketInitializer
21 __global__ void InitKernel(const int env_num, const int agent_num, const GameSetting *setting, in InitKernel() function