| /frameworks/ml/nn/runtime/test/specs/V1_0/ |
| 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]
|
| 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_1/ |
| 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]
|
| 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]
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| svdf_test.cc | 229 float* batch_start = svdf_input + i * svdf_input_size * svdf_num_batches; local
230 float* batch_end = batch_start + svdf_input_size * svdf_num_batches;
231 svdf.SetInput(0, batch_start, batch_end);
289 float* batch_start = svdf_input + i * svdf_input_size * svdf_num_batches; local
290 float* batch_end = batch_start + svdf_input_size * svdf_num_batches;
291 svdf.SetInput(0, batch_start, batch_end);
|
| unidirectional_sequence_rnn_test.cc | 257 float* batch_start = rnn_input; local
258 float* batch_end = batch_start + input_sequence_size;
259 rnn.SetInput(0, batch_start, batch_end);
260 rnn.SetInput(input_sequence_size, batch_start, batch_end);
324 float* batch_start = rnn_input + i * rnn.input_size(); local
325 float* batch_end = batch_start + rnn.input_size();
327 rnn.SetInput(2 * i * rnn.input_size(), batch_start, batch_end);
328 rnn.SetInput((2 * i + 1) * rnn.input_size(), batch_start, batch_end);
|
| basic_rnn_test.cc | 242 float* batch_start = rnn_input + i * rnn.input_size(); local
243 float* batch_end = batch_start + rnn.input_size();
244 rnn.SetInput(0, batch_start, batch_end);
245 rnn.SetInput(rnn.input_size(), batch_start, batch_end);
|
| fully_connected_test.cc | 352 float* batch_start = fully_connected_input + i * m.input_size(); local
353 float* batch_end = batch_start + m.input_size();
354 m.SetInput(0, batch_start, batch_end);
355 m.SetInput(m.input_size(), batch_start, batch_end);
|
| bidirectional_sequence_rnn_test.cc | 779 float* batch_start = rnn_input; local
821 float* batch_start = rnn_input + i * rnn.input_size(); local
889 float* batch_start = endtoend_input + k * input_sequence_size; local
[all...] |
| /frameworks/ml/nn/common/operations/ |
| SVDFTest.cpp | 369 float* batch_start = svdf_input + i * svdf_input_size * svdf_num_batches; local
370 float* batch_end = batch_start + svdf_input_size * svdf_num_batches;
371 svdf.SetInput(0, batch_start, batch_end);
431 float* batch_start = svdf_input_rank2 + i * svdf_input_size * svdf_num_batches; local
432 float* batch_end = batch_start + svdf_input_size * svdf_num_batches;
433 svdf.SetInput(0, batch_start, batch_end);
|
| RNNTest.cpp | 318 float* batch_start = rnn_input + i * rnn.input_size(); local
319 float* batch_end = batch_start + rnn.input_size();
320 rnn.SetInput(0, batch_start, batch_end);
321 rnn.SetInput(rnn.input_size(), batch_start, batch_end);
|
| /external/libtextclassifier/ |
| text-classifier.cc | [all...] |
| /external/tensorflow/tensorflow/python/keras/_impl/keras/engine/ |
| training_eager.py | 395 for batch_index, (batch_start, batch_end) in enumerate(batches):
396 batch_ids = index_array[batch_start:batch_end]
508 for batch_index, (batch_start, batch_end) in enumerate(batches):
509 batch_ids = index_array[batch_start:batch_end]
586 for batch_index, (batch_start, batch_end) in enumerate(batches):
587 batch_ids = index_array[batch_start:batch_end]
623 outs[i][batch_start:batch_end] = batch_out
|
| training.py | [all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| deep_conv2d.cc | [all...] |
| /external/tensorflow/tensorflow/contrib/metrics/python/ops/ |
| metric_ops_test.py | [all...] |
