Searched refs:HashtableLookup (Results 1 – 5 of 5) sorted by relevance
36 HashtableLookup::HashtableLookup(const Operation& operation, RunTimeOperandInfo* operands) { in HashtableLookup() function in android::nn::HashtableLookup45 bool HashtableLookup::Eval() { in Eval()
118 ASSERT_EQ(execution.setInput(HashtableLookup::k##X##Tensor, X##_.data(), \ in Invoke()127 ASSERT_EQ(execution.setOutput(HashtableLookup::k##X##Tensor, X##_.data(), \ in Invoke()
30 class HashtableLookup {32 HashtableLookup(const Operation& operation, RunTimeOperandInfo* operands);
155 "cpu_operations/HashtableLookup.cpp",342 "cpu_operations/HashtableLookup.cpp",
958 const RunTimeOperandInfo& lookups = operands[ins[HashtableLookup::kLookupTensor]]; in executeOperation()959 const RunTimeOperandInfo& keys = operands[ins[HashtableLookup::kKeyTensor]]; in executeOperation()960 const RunTimeOperandInfo& values = operands[ins[HashtableLookup::kValueTensor]]; in executeOperation()962 RunTimeOperandInfo& output = operands[outs[HashtableLookup::kOutputTensor]]; in executeOperation()963 RunTimeOperandInfo& hits = operands[outs[HashtableLookup::kHitsTensor]]; in executeOperation()966 HashtableLookup lookup(operation, operands); in executeOperation()