Home
last modified time | relevance | path

Searched defs:batch_start (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_rnn_test.cc287 float* batch_start = rnn_input; in TEST() local
316 float* batch_start = rnn_input; in TEST_P() local
343 float* batch_start = rnn_input; in TEST_P() local
369 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
399 float* batch_start = rnn_input + i * rnn.input_size(); in TEST_P() local
430 float* batch_start = rnn_input + i * rnn.input_size(); in TEST_P() local
Dbidirectional_sequence_rnn_test.cc878 float* batch_start = rnn_input; in TEST_P() local
929 float* batch_start = rnn_input + i * rnn.input_size(); in TEST_P() local
973 float* batch_start = rnn_input; in TEST_P() local
1017 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
1062 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
1131 float* batch_start = endtoend_input + k * input_sequence_size; in TEST() local
1188 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
1240 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
1288 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
1332 float* batch_start = rnn_input; in TEST() local
[all …]
Dbasic_rnn_test.cc269 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
298 float* batch_start = rnn_input + i * rnn.input_size(); in TEST_P() local
326 float* batch_start = rnn_input + i * rnn.input_size(); in TEST_P() local
Dsvdf_test.cc247 float* batch_start = in VerifyGoldens() local
Dunidirectional_sequence_lstm_test.cc402 const float* batch_start = input[b].data() + i * num_inputs; in VerifyGoldens() local
411 const float* batch_start = input[b].data(); in VerifyGoldens() local
2524 const float* batch_start = input[b].data() + i * num_inputs; in VerifyGoldens() local
Dlstm_test.cc359 const float* batch_start = lstm_input_[i][b].data(); in VerifyGoldens() local
370 const float* batch_start = lstm_golden_output_[i][b].data(); in VerifyGoldens() local
2400 const float* batch_start = input[b].data() + i * num_inputs; in VerifyGoldens() local
Dfully_connected_test.cc1165 float* batch_start = fully_connected_input + i * m.input_size(); in TEST_P() local
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils_v1.py105 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
143 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
286 def aggregate(self, batch_element, batch_start=None, batch_end=None): argument
377 def aggregate(self, batch_element, batch_start, batch_end): argument
405 def _slice_assign(self, batch_element, batch_start, batch_end, is_finished): argument
461 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
Dtraining_utils_v1_test.py265 …def wrapped(batch_element, batch_start, batch_end, is_finished): # pylint: disable=unused-argument argument
/external/rust/crates/plotters-backend/src/rasterizer/
Dline.rs94 let batch_start = (f64::from(from.1.min(size_limit.1 as i32 - 2).max(0) - from.1) / grad) in draw_line() localVariable
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
Dlstm_full_test.cc242 const float* batch_start = lstm_input_[i][b].data(); in VerifyGoldens() local
253 const float* batch_start = lstm_golden_output_[i][b].data(); in VerifyGoldens() local
/external/XNNPACK/src/
Doperator-run.c721 size_t batch_start, in xnn_compute_prelu()
880 size_t batch_start, in xnn_compute_vmulcaddc()
Dindirection.c524 size_t batch_start, in xnn_indirection_init_unpool2d()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h144 int batch_start = 0; in DepthwiseConvHybridGeneral() local
/external/libtextclassifier/native/annotator/
Dannotator.cc2987 for (int batch_start = span_of_interest.first; in ModelClickContextScoreChunks() local
3096 for (int batch_start = 0; batch_start < candidate_spans.size(); in ModelBoundsSensitiveScoreChunks() local
/external/tensorflow/tensorflow/core/kernels/
Ddeep_conv2d.cc1009 &output](int64 batch_start, int64 batch_limit) { in operator ()()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2804 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
3007 float* batch_start = in VerifyGoldens() local
3384 const float* batch_start = input[b].data() + i * num_inputs; in VerifyGoldens() local
/external/igt-gpu-tools/benchmarks/
Dgem_wsim.c1118 unsigned long batch_start = w->bb_sz; in terminate_bb() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h3724 int batch_start = 0; variable