Searched refs:multiAll (Results 1 – 5 of 5) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | HashtableLookupTest.cpp | 89 auto multiAll = [](const std::vector<uint32_t> &dims) -> uint32_t { in HashtableLookupOpModel() local
95 Value_.insert(Value_.end(), multiAll(value_shape), 0.f); in HashtableLookupOpModel()
96 Output_.insert(Output_.end(), multiAll(out_dim), 0.f); in HashtableLookupOpModel()
97 Hits_.insert(Hits_.end(), multiAll(lookup_shape), 0); in HashtableLookupOpModel()
|
| D | EmbeddingLookupTest.cpp | 76 auto multiAll = [](const std::vector<uint32_t> &dims) -> uint32_t { in EmbeddingLookupOpModel() local
82 Value_.insert(Value_.end(), multiAll(weight_shape), 0.f); in EmbeddingLookupOpModel()
83 Output_.insert(Output_.end(), multiAll(weight_shape), 0.f); in EmbeddingLookupOpModel()
|
| D | LSHProjectionTest.cpp | 65 auto multiAll = [](const std::vector<uint32_t> &dims) -> uint32_t { in LSHProjectionOpModel() local
77 Output_.insert(Output_.end(), multiAll(hash_shape), 0.f); in LSHProjectionOpModel()
|
| D | SVDFTest.cpp | 172 auto multiAll = [](const std::vector<uint32_t> &dims) -> uint32_t { in SVDFOpModel() local
180 #define ReserveOutput(X) X##_.insert(X##_.end(), multiAll(*it2++), 0.f); in SVDFOpModel()
|
| D | LSTMTest.cpp | 138 auto multiAll = [](const std::vector<uint32_t> &dims) -> uint32_t { in LSTMOpModel() local
146 #define ReserveOutput(X) X##_.insert(X##_.end(), multiAll(*it2++), 0.f); in LSTMOpModel()
|