Home
last modified time | relevance | path

Searched defs:batch_start (Results 1 – 22 of 22) 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
Dunidirectional_sequence_lstm_test.cc150 const float* batch_start = input[b].data() + i * num_inputs; in VerifyGoldens() local
159 const float* batch_start = input[b].data(); in VerifyGoldens() local
2272 const float* batch_start = input[b].data() + i * num_inputs; in VerifyGoldens() local
Dsvdf_test.cc247 float* batch_start = 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.cc1521 float* batch_start = fully_connected_input + i * m.input_size(); in TEST_P() local
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_utils_v1.py99 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
137 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
280 def aggregate(self, batch_element, batch_start=None, batch_end=None): argument
371 def aggregate(self, batch_element, batch_start, batch_end): argument
399 def _slice_assign(self, batch_element, batch_start, batch_end, is_finished): argument
455 def aggregate(self, batch_outs, batch_start=None, batch_end=None): argument
/external/rust/android-crates-io/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/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
Dindirection.c226 size_t batch_start) { in pytorch_qnnp_indirection_init_dwconv()
410 size_t batch_start) { in pytorch_qnnp_indirection_init_maxpool2d()
/external/mesa3d/src/intel/executor/
Dexecutor.h44 uint64_t batch_start; member
/external/pytorch/aten/src/ATen/native/cuda/
DEmbedding.cu65 for(int batch_start = 0; batch_start < n; batch_start += blockDim.x*blockDim.y) in embedding_backward_feature_kernel() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h146 int batch_start = 0; in DepthwiseConvHybridGeneral() local
/external/XNNPACK/src/
Doperator-run.c937 size_t batch_start, in xnn_compute_prelu()
1159 size_t batch_start, in xnn_compute_vmulcaddc()
Dindirection.c826 size_t batch_start, in xnn_indirection_init_unpool2d()
/external/libtextclassifier/native/annotator/
Dannotator.cc3004 for (int batch_start = span_of_interest.first; in ModelClickContextScoreChunks() local
3113 for (int batch_start = 0; batch_start < candidate_spans.size(); in ModelBoundsSensitiveScoreChunks() local
/external/tensorflow/tensorflow/core/kernels/
Ddeep_conv2d.cc1023 &output](int64_t batch_start, int64_t batch_limit) { in operator ()()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2881 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
3084 float* batch_start = in VerifyGoldens() local
3461 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/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
DQuantizedOpKernels.cpp1838 at::parallel_for(0, nBatch, 0, [&](int64_t batch_start, int64_t batch_end) {
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h3408 int batch_start = 0; variable