Searched refs:last_dim_size_ (Results 1 – 5 of 5) sorted by relevance
40 parameter->last_dim_size_ = input->shape().at(input->shape().size() - 1); in ReSize()58 …topk_node_list_ = ms_context_->allocator->Malloc(sizeof(TopkNodeInt8) * parameter->last_dim_size_); in Run()
39 topk_param_->last_dim_size_ = input->shape().at(input->shape().size() - 1); in ReSize()66 … ms_context_->allocator->Malloc(static_cast<int>(sizeof(TopkNode)) * topk_param_->last_dim_size_); in Run()
34 int last_dim_size_; member
42 int last_dim_size = parameter->last_dim_size_; in Topk()
28 int last_dim_size = parameter->last_dim_size_; in TopkInt8()