HomeSort by relevance Sort by last modified time
    Searched refs:lookup_ (Results 1 - 4 of 4) sorted by null

  /frameworks/ml/nn/common/operations/
EmbeddingLookup.cpp 29 lookup_ = GetInput(operation, operands, kLookupTensor);
39 for (uint32_t i = 0; i < lookup_->shape().dimensions[0]; i++) {
40 int idx = (reinterpret_cast<int*>(lookup_->buffer))[i];
EmbeddingLookup.h 52 const RunTimeOperandInfo *lookup_; member in class:android::nn::EmbeddingLookup
HashtableLookup.cpp 36 lookup_ = GetInput(operation, operands, kLookupTensor);
49 for (int i = 0; i < static_cast<int>(lookup_->shape().dimensions[0]); i++) {
51 pointer = bsearch(lookup_->buffer + sizeof(int) * i, key_->buffer,
HashtableLookup.h 53 const RunTimeOperandInfo *lookup_; member in class:android::nn::HashtableLookup

Completed in 227 milliseconds