Searched refs:EmbeddingLookup (Results 1 - 4 of 4) sorted by relevance

/frameworks/ml/nn/common/operations/
H A DEmbeddingLookup.cpp17 #include "EmbeddingLookup.h"
26 EmbeddingLookup::EmbeddingLookup(const Operation& operation, function in class:android::nn::EmbeddingLookup
34 bool EmbeddingLookup::Eval() {
H A DEmbeddingLookup.h37 class EmbeddingLookup { class in namespace:android::nn
39 EmbeddingLookup(
H A DEmbeddingLookupTest.cpp17 #include "EmbeddingLookup.h"
99 ASSERT_EQ(execution.setInput(EmbeddingLookup::k##X##Tensor, X##_.data(), \
108 ASSERT_EQ(execution.setOutput(EmbeddingLookup::k##X##Tensor, X##_.data(), \
/frameworks/ml/nn/common/
H A DCpuExecutor.cpp1210 mOperands[ins[EmbeddingLookup::kValueTensor]];
1212 mOperands[ins[EmbeddingLookup::kLookupTensor]];
1214 mOperands[outs[EmbeddingLookup::kOutputTensor]];
1217 EmbeddingLookup lookup(operation, mOperands);

Completed in 211 milliseconds