Searched refs:batch_end (Results 1 – 11 of 11) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | SVDFTest.cpp | 347 float* batch_end = batch_start + svdf_input_size * svdf_num_batches; in TEST() local
348 svdf.SetInput(0, batch_start, batch_end); in TEST()
406 float* batch_end = batch_start + svdf_input_size * svdf_num_batches; in TEST() local
407 svdf.SetInput(0, batch_start, batch_end); in TEST()
|
| D | RNNTest.cpp | 288 float* batch_end = batch_start + rnn.input_size(); in TEST() local
289 rnn.SetInput(0, batch_start, batch_end); in TEST()
290 rnn.SetInput(rnn.input_size(), batch_start, batch_end); in TEST()
|
| D | LayerNormLSTMTest.cpp | 409 const float* batch_end = batch_start + n_input; in TEST() local
411 lstm.SetInput(b * n_input, batch_start, batch_end); in TEST()
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_0/ |
| D | svdf.mod.py | 133 batch_end = batch_start + input_size * batches variable
134 input0[input] = test_inputs[batch_start:batch_end]
|
| D | svdf_bias_present.mod.py | 133 batch_end = batch_start + input_size * batches variable
134 input0[input] = test_inputs[batch_start:batch_end]
|
| D | svdf2.mod.py | 148 batch_end = batch_start + input_size * batches variable
149 input0[input] = test_inputs[batch_start:batch_end]
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | svdf_bias_present_float16.mod.py | 133 batch_end = batch_start + input_size * batches variable
134 input0[input] = test_inputs[batch_start:batch_end]
|
| D | svdf_float16.mod.py | 133 batch_end = batch_start + input_size * batches variable
134 input0[input] = test_inputs[batch_start:batch_end]
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_1/ |
| D | svdf2_relaxed.mod.py | 149 batch_end = batch_start + input_size * batches variable
150 input0[input] = test_inputs[batch_start:batch_end]
|
| D | svdf_relaxed.mod.py | 134 batch_end = batch_start + input_size * batches variable
135 input0[input] = test_inputs[batch_start:batch_end]
|
| D | svdf_bias_present_relaxed.mod.py | 134 batch_end = batch_start + input_size * batches variable
135 input0[input] = test_inputs[batch_start:batch_end]
|