Searched refs:ids_size_ (Results 1 – 3 of 3) sorted by relevance
29 int ids_size_; member
55 for (int i = task_id; i < parameter->ids_size_; i += parameter->op_parameter_.thread_num_) { in EmbeddingLookup()
38 param_->ids_size_ = in_tensors_.back()->ElementsNum(); in ReSize()