Searched refs:EmbeddingLookup (Results 1 – 5 of 5) sorted by relevance
28 EmbeddingLookup::EmbeddingLookup(const Operation& operation, RunTimeOperandInfo* operands) { in EmbeddingLookup() function in android::nn::EmbeddingLookup35 bool EmbeddingLookup::Eval() { in Eval()
102 ASSERT_EQ(execution.setInput(EmbeddingLookup::k##X##Tensor, X##_.data(), \ in Invoke()111 ASSERT_EQ(execution.setOutput(EmbeddingLookup::k##X##Tensor, X##_.data(), \ in Invoke()
30 class EmbeddingLookup {32 EmbeddingLookup(const Operation& operation, RunTimeOperandInfo* operands);
152 "cpu_operations/EmbeddingLookup.cpp",338 "cpu_operations/EmbeddingLookup.cpp",
944 const RunTimeOperandInfo& values = operands[ins[EmbeddingLookup::kValueTensor]]; in executeOperation()945 const RunTimeOperandInfo& lookups = operands[ins[EmbeddingLookup::kLookupTensor]]; in executeOperation()946 RunTimeOperandInfo& output = operands[outs[EmbeddingLookup::kOutputTensor]]; in executeOperation()949 EmbeddingLookup lookup(operation, operands); in executeOperation()