Searched refs:topk_node_list_ (Results 1 – 6 of 6) sorted by relevance
30 topk_param_->topk_node_list_ = nullptr; in Init()65 topk_param_->topk_node_list_ = in Run()67 if (topk_param_->topk_node_list_ == nullptr) { in Run()72 ms_context_->allocator->Free(topk_param_->topk_node_list_); in Run()73 topk_param_->topk_node_list_ = nullptr; in Run()
58 …parameter->topk_node_list_ = ms_context_->allocator->Malloc(sizeof(TopkNodeInt8) * parameter->last… in Run()59 if (parameter->topk_node_list_ == nullptr) { in Run()64 ms_context_->allocator->Free(parameter->topk_node_list_); in Run()
30 param->topk_node_list_ = nullptr; in TopKInt8CPUKernel()
36 void *topk_node_list_; member
45 TopkNode *top_map = (TopkNode *)parameter->topk_node_list_; in Topk()
31 TopkNodeInt8 *top_map = (TopkNodeInt8 *)parameter->topk_node_list_; in TopkInt8()