Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DLSHProjectionTest.cpp142 m.SetInput({12345, 54321, 67890, 9876, -12345678, -87654321}); in TEST()
154 m.SetInput({12345, 54321, 67890, 9876, -12345678, -87654321}); in TEST()
165 m.SetInput({12345, 54321, 67890, 9876, -12345678, -87654321}); in TEST()
DLSTMTest.cpp177 void SetInput(int offset, float* begin, float* end) { in SetInput() function in android::nn::wrapper::LSTMOpModel
364 lstm.SetInput(0, batch0_start, batch0_end); in TEST()
461 lstm.SetInput(0, batch0_start, batch0_end); in TEST()
989 lstm.SetInput(0, batch0_start, batch0_end); in TEST()
993 lstm.SetInput(lstm.num_inputs(), batch1_start, batch1_end); in TEST()
DRNNTest.cpp175 void SetInput(int offset, float* begin, float* end) { in SetInput() function in android::nn::wrapper::BasicRNNOpModel
289 rnn.SetInput(0, batch_start, batch_end); in TEST()
290 rnn.SetInput(rnn.input_size(), batch_start, batch_end); in TEST()
DSVDFTest.cpp280 void SetInput(int offset, float* begin, float* end) { in SetInput() function in android::nn::wrapper::SVDFOpModel
348 svdf.SetInput(0, batch_start, batch_end); in TEST()
407 svdf.SetInput(0, batch_start, batch_end); in TEST()
DLayerNormLSTMTest.cpp189 void SetInput(int offset, const float* begin, const float* end) { in SetInput() function in android::nn::wrapper::LayerNormLSTMOpModel
411 lstm.SetInput(b * n_input, batch_start, batch_end); in TEST()
/packages/modules/NeuralNetworks/runtime/test/
DTestValidation.cpp1603 TEST_F(ValidationTestExecution, SetInput) { in TEST_F() argument