Searched refs:lookup_ (Results 1 – 4 of 4) sorted by relevance
30 lookup_ = GetInput(operation, operands, kLookupTensor); in EmbeddingLookup()41 for (uint32_t i = 0; i < lookup_->shape().dimensions[0]; i++) { in Eval()42 int idx = (reinterpret_cast<int*>(lookup_->buffer))[i]; in Eval()
37 lookup_ = GetInput(operation, operands, kLookupTensor); in HashtableLookup()52 for (int i = 0; i < static_cast<int>(lookup_->shape().dimensions[0]); i++) { in Eval()54 pointer = bsearch(lookup_->buffer + sizeof(int) * i, key_->buffer, num_rows, sizeof(int), in Eval()
43 const RunTimeOperandInfo* lookup_; variable
44 const RunTimeOperandInfo* lookup_;