Home
last modified time | relevance | path

Searched refs:kLookupTensor (Results 1 – 5 of 5) sorted by relevance

/frameworks/ml/nn/common/operations/
DEmbeddingLookup.h35 static constexpr int kLookupTensor = 0; variable
DHashtableLookup.h35 static constexpr int kLookupTensor = 0; variable
DEmbeddingLookup.cpp34 lookup_ = GetInput(operation, operands, kLookupTensor); in EmbeddingLookup()
DHashtableLookup.cpp41 lookup_ = GetInput(operation, operands, kLookupTensor); in HashtableLookup()
/frameworks/ml/nn/common/
DCpuExecutor.cpp998 const RunTimeOperandInfo& lookups = operands[ins[EmbeddingLookup::kLookupTensor]]; in executeOperation()
1011 const RunTimeOperandInfo& lookups = operands[ins[HashtableLookup::kLookupTensor]]; in executeOperation()
1114 const RunTimeOperandInfo& lookups = operands[ins[HashtableLookup::kLookupTensor]]; in executeOperation()