Home
last modified time | relevance | path

Searched refs:batch_end (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/runtime/test/specs/
Dsvdf.mod.py132 batch_end = batch_start + input_size * batches variable
133 input0[input] = test_inputs[batch_start:batch_end]
/frameworks/ml/nn/common/operations/
DRNNTest.cpp319 float* batch_end = batch_start + rnn.input_size(); in TEST() local
320 rnn.SetInput(0, batch_start, batch_end); in TEST()
321 rnn.SetInput(rnn.input_size(), batch_start, batch_end); in TEST()
DSVDFTest.cpp306 float* batch_end = batch_start + svdf_input_size * svdf_num_batches; in TEST() local
307 svdf.SetInput(0, batch_start, batch_end); in TEST()