Home
last modified time | relevance | path

Searched refs:DefineTensor (Results 1 – 6 of 6) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DEmbeddingLookupTest.cpp147 #define DefineTensor(X, T) std::vector<T> X##_; macro
149 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(DefineTensor);
150 FOR_ALL_OUTPUT_TENSORS(DefineTensor);
152 #undef DefineTensor
H A DHashtableLookupTest.cpp160 #define DefineTensor(X, T) std::vector<T> X##_; macro
162 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(DefineTensor);
163 FOR_ALL_OUTPUT_TENSORS(DefineTensor);
165 #undef DefineTensor
H A DRNNTest.cpp231 #define DefineTensor(X) std::vector<float> X##_; macro
233 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(DefineTensor);
234 FOR_ALL_OUTPUT_TENSORS(DefineTensor);
236 #undef DefineTensor
H A DLayerNormLSTMTest.cpp276 #define DefineTensor(X) std::vector<float> X##_; macro
278 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(DefineTensor);
279 FOR_ALL_LAYER_NORM_WEIGHTS(DefineTensor);
280 FOR_ALL_OUTPUT_TENSORS(DefineTensor);
282 #undef DefineTensor
H A DSVDFTest.cpp308 #define DefineTensor(X) std::vector<float> X##_; macro
310 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(DefineTensor);
311 FOR_ALL_OUTPUT_TENSORS(DefineTensor);
313 #undef DefineTensor
H A DLSTMTest.cpp264 #define DefineTensor(X) std::vector<float> X##_; macro
266 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(DefineTensor);
267 FOR_ALL_OUTPUT_TENSORS(DefineTensor);
269 #undef DefineTensor