Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DEmbeddingLookup.h36 static constexpr int kValueTensor = 1; variable
H A DHashtableLookup.h37 static constexpr int kValueTensor = 2; variable
H A DEmbeddingLookup.cpp29 value_ = GetInput(operation, operands, kValueTensor); in EmbeddingLookup()
H A DHashtableLookup.cpp39 value_ = GetInput(operation, operands, kValueTensor); in HashtableLookup()
/aosp12/packages/modules/NeuralNetworks/common/
H A DCpuExecutor.cpp943 const RunTimeOperandInfo& values = operands[ins[EmbeddingLookup::kValueTensor]]; in executeOperation()
959 const RunTimeOperandInfo& values = operands[ins[HashtableLookup::kValueTensor]]; in executeOperation()