Searched refs:EmbeddingLookup (Results 1 – 5 of 5) sorted by relevance
32 EmbeddingLookup::EmbeddingLookup(const Operation& operation, RunTimeOperandInfo* operands) { in EmbeddingLookup() function in android::nn::EmbeddingLookup39 bool EmbeddingLookup::Eval() { in Eval()
29 class EmbeddingLookup {31 EmbeddingLookup(const hal::Operation& operation, RunTimeOperandInfo* operands);
101 ASSERT_EQ(execution.setInput(EmbeddingLookup::k##X##Tensor, X##_.data(), \ in Invoke()110 ASSERT_EQ(execution.setOutput(EmbeddingLookup::k##X##Tensor, X##_.data(), \ in Invoke()
163 "operations/EmbeddingLookup.cpp",
997 const RunTimeOperandInfo& values = operands[ins[EmbeddingLookup::kValueTensor]]; in executeOperation()998 const RunTimeOperandInfo& lookups = operands[ins[EmbeddingLookup::kLookupTensor]]; in executeOperation()999 RunTimeOperandInfo& output = operands[outs[EmbeddingLookup::kOutputTensor]]; in executeOperation()1002 EmbeddingLookup lookup(operation, operands); in executeOperation()