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