Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DLSHProjectionTest.cpp105 #define SetInputOrWeight(X, T) \ in Invoke() macro
110 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
112 #undef SetInputOrWeight in Invoke()
DEmbeddingLookupTest.cpp102 #define SetInputOrWeight(X, T) \ in Invoke() macro
107 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
109 #undef SetInputOrWeight in Invoke()
DHashtableLookupTest.cpp117 #define SetInputOrWeight(X, T) \ in Invoke() macro
122 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
124 #undef SetInputOrWeight in Invoke()
DRNNTest.cpp200 #define SetInputOrWeight(X) \ in Invoke() macro
204 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
206 #undef SetInputOrWeight in Invoke()
DLayerNormLSTMTest.cpp209 #define SetInputOrWeight(X) \ in Invoke() macro
214 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
215 FOR_ALL_LAYER_NORM_WEIGHTS(SetInputOrWeight); in Invoke()
217 #undef SetInputOrWeight in Invoke()
DSVDFTest.cpp247 #define SetInputOrWeight(X) \ in Invoke() macro
251 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
253 #undef SetInputOrWeight in Invoke()
DLSTMTest.cpp197 #define SetInputOrWeight(X) \ in Invoke() macro
202 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
204 #undef SetInputOrWeight in Invoke()