Home
last modified time | relevance | path

Searched refs:SetInputOrWeight (Results 1 – 7 of 7) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DEmbeddingLookupTest.cpp101 #define SetInputOrWeight(X, T) \ in Invoke() macro
106 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
108 #undef SetInputOrWeight in Invoke()
H A DLSHProjectionTest.cpp105 #define SetInputOrWeight(X, T) \ in Invoke() macro
110 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
112 #undef SetInputOrWeight in Invoke()
H A DHashtableLookupTest.cpp116 #define SetInputOrWeight(X, T) \ in Invoke() macro
121 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
123 #undef SetInputOrWeight in Invoke()
H A DRNNTest.cpp200 #define SetInputOrWeight(X) \ in Invoke() macro
204 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
206 #undef SetInputOrWeight in Invoke()
H A DLayerNormLSTMTest.cpp208 #define SetInputOrWeight(X) \ in Invoke() macro
213 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
214 FOR_ALL_LAYER_NORM_WEIGHTS(SetInputOrWeight); in Invoke()
216 #undef SetInputOrWeight in Invoke()
H A DSVDFTest.cpp247 #define SetInputOrWeight(X) \ in Invoke() macro
251 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
253 #undef SetInputOrWeight in Invoke()
H A DLSTMTest.cpp197 #define SetInputOrWeight(X) \ in Invoke() macro
202 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
204 #undef SetInputOrWeight in Invoke()