Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DHashtableLookup.cpp40 HashtableLookup::HashtableLookup(const Operation& operation, RunTimeOperandInfo* operands) { in HashtableLookup() function in android::nn::HashtableLookup
49 bool HashtableLookup::Eval() { in Eval()
DHashtableLookup.h29 class HashtableLookup {
31 HashtableLookup(const hal::Operation& operation, RunTimeOperandInfo* operands);
DHashtableLookupTest.cpp116 ASSERT_EQ(execution.setInput(HashtableLookup::k##X##Tensor, X##_.data(), \ in Invoke()
125 ASSERT_EQ(execution.setOutput(HashtableLookup::k##X##Tensor, X##_.data(), \ in Invoke()
/frameworks/ml/nn/common/
DCpuExecutor.cpp1013 const RunTimeOperandInfo& lookups = operands[ins[HashtableLookup::kLookupTensor]]; in executeOperation()
1014 const RunTimeOperandInfo& keys = operands[ins[HashtableLookup::kKeyTensor]]; in executeOperation()
1015 const RunTimeOperandInfo& values = operands[ins[HashtableLookup::kValueTensor]]; in executeOperation()
1017 RunTimeOperandInfo& output = operands[outs[HashtableLookup::kOutputTensor]]; in executeOperation()
1018 RunTimeOperandInfo& hits = operands[outs[HashtableLookup::kHitsTensor]]; in executeOperation()
1021 HashtableLookup lookup(operation, operands); in executeOperation()
DAndroid.bp166 "operations/HashtableLookup.cpp",