Searched refs:batch_start (Results 1 – 11 of 11) sorted by relevance
| /frameworks/ml/nn/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]
|
| /frameworks/ml/nn/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]
|
| /frameworks/ml/nn/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]
|
| /frameworks/ml/nn/common/operations/ |
| D | SVDFTest.cpp | 344 float* batch_start = svdf_input + i * svdf_input_size * svdf_num_batches; in TEST() local
345 float* batch_end = batch_start + svdf_input_size * svdf_num_batches; in TEST()
346 svdf.SetInput(0, batch_start, batch_end); in TEST()
403 float* batch_start = svdf_input_rank2 + i * svdf_input_size * svdf_num_batches; in TEST() local
404 float* batch_end = batch_start + svdf_input_size * svdf_num_batches; in TEST()
405 svdf.SetInput(0, batch_start, batch_end); in TEST()
|
| D | RNNTest.cpp | 286 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
287 float* batch_end = batch_start + rnn.input_size(); in TEST()
288 rnn.SetInput(0, batch_start, batch_end); in TEST()
289 rnn.SetInput(rnn.input_size(), batch_start, batch_end); in TEST()
|
| D | LayerNormLSTMTest.cpp | 407 const float* batch_start = lstm_input[b].data() + i * n_input; in TEST() local
408 const float* batch_end = batch_start + n_input; in TEST()
410 lstm.SetInput(b * n_input, batch_start, batch_end); in TEST()
|