Searched refs:EmbeddingLookup (Results 1 – 5 of 5) sorted by relevance
27 EmbeddingLookup::EmbeddingLookup(const Operation& operation, in EmbeddingLookup() function in android::nn::EmbeddingLookup35 bool EmbeddingLookup::Eval() { in Eval()
29 class EmbeddingLookup {31 EmbeddingLookup(
99 ASSERT_EQ(execution.setInput(EmbeddingLookup::k##X##Tensor, X##_.data(), \ in Invoke()108 ASSERT_EQ(execution.setOutput(EmbeddingLookup::k##X##Tensor, X##_.data(), \ in Invoke()
133 "operations/EmbeddingLookup.cpp",
1049 const RunTimeOperandInfo& values = mOperands[ins[EmbeddingLookup::kValueTensor]]; in executeOperation()1050 const RunTimeOperandInfo& lookups = mOperands[ins[EmbeddingLookup::kLookupTensor]]; in executeOperation()1051 RunTimeOperandInfo& output = mOperands[outs[EmbeddingLookup::kOutputTensor]]; in executeOperation()1054 EmbeddingLookup lookup(operation, mOperands); in executeOperation()