Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DEmbeddingLookupTest.cpp98 #define SetInputOrWeight(X, T) \ in Invoke() macro
103 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
105 #undef SetInputOrWeight in Invoke()
DLSHProjectionTest.cpp104 #define SetInputOrWeight(X, T) \ in Invoke() macro
109 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
111 #undef SetInputOrWeight in Invoke()
DHashtableLookupTest.cpp112 #define SetInputOrWeight(X, T) \ in Invoke() macro
117 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
119 #undef SetInputOrWeight in Invoke()
DRNNTest.cpp223 #define SetInputOrWeight(X) \ in Invoke() macro
228 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
230 #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.cpp263 #define SetInputOrWeight(X) \ in Invoke() macro
268 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
270 #undef SetInputOrWeight in Invoke()
DLSTMTest.cpp194 #define SetInputOrWeight(X) \ in Invoke() macro
199 FOR_ALL_INPUT_AND_WEIGHT_TENSORS(SetInputOrWeight); in Invoke()
201 #undef SetInputOrWeight in Invoke()