Searched refs:batch_start (Results 1 – 11 of 11) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | SVDFTest.cpp | 346 float* batch_start = svdf_input + i * svdf_input_size * svdf_num_batches; in TEST() local
347 float* batch_end = batch_start + svdf_input_size * svdf_num_batches; in TEST()
348 svdf.SetInput(0, batch_start, batch_end); in TEST()
405 float* batch_start = svdf_input_rank2 + i * svdf_input_size * svdf_num_batches; in TEST() local
406 float* batch_end = batch_start + svdf_input_size * svdf_num_batches; in TEST()
407 svdf.SetInput(0, batch_start, batch_end); in TEST()
|
| D | RNNTest.cpp | 287 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
288 float* batch_end = batch_start + rnn.input_size(); in TEST()
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 | 408 const float* batch_start = lstm_input[b].data() + i * n_input; in TEST() local
409 const float* batch_end = batch_start + n_input; in TEST()
411 lstm.SetInput(b * n_input, batch_start, batch_end); in TEST()
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_0/ |
| D | svdf.mod.py | 132 batch_start = i * input_size * batches variable
133 batch_end = batch_start + input_size * batches
134 input0[input] = test_inputs[batch_start:batch_end]
|
| D | svdf_bias_present.mod.py | 132 batch_start = i * input_size * batches variable
133 batch_end = batch_start + input_size * batches
134 input0[input] = test_inputs[batch_start:batch_end]
|
| D | svdf2.mod.py | 147 batch_start = i * input_size * batches variable
148 batch_end = batch_start + input_size * batches
149 input0[input] = test_inputs[batch_start:batch_end]
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_2/ |
| D | svdf_bias_present_float16.mod.py | 132 batch_start = i * input_size * batches variable
133 batch_end = batch_start + input_size * batches
134 input0[input] = test_inputs[batch_start:batch_end]
|
| D | svdf_float16.mod.py | 132 batch_start = i * input_size * batches variable
133 batch_end = batch_start + input_size * batches
134 input0[input] = test_inputs[batch_start:batch_end]
|
| /packages/modules/NeuralNetworks/runtime/test/specs/V1_1/ |
| D | svdf2_relaxed.mod.py | 148 batch_start = i * input_size * batches variable
149 batch_end = batch_start + input_size * batches
150 input0[input] = test_inputs[batch_start:batch_end]
|
| D | svdf_relaxed.mod.py | 133 batch_start = i * input_size * batches variable
134 batch_end = batch_start + input_size * batches
135 input0[input] = test_inputs[batch_start:batch_end]
|
| D | svdf_bias_present_relaxed.mod.py | 133 batch_start = i * input_size * batches variable
134 batch_end = batch_start + input_size * batches
135 input0[input] = test_inputs[batch_start:batch_end]
|