Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A 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 …]
H A 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
H A 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
H A 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
H A 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
H A Dsvdf_test.cc247 float* batch_start = in VerifyGoldens() local
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A 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
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DSVDFTest.cpp346 float* batch_start = svdf_input + i * svdf_input_size * svdf_num_batches; in TEST() local
405 float* batch_start = svdf_input_rank2 + i * svdf_input_size * svdf_num_batches; in TEST() local
DRNNTest.cpp287 float* batch_start = rnn_input + i * rnn.input_size(); in TEST() local
DLayerNormLSTMTest.cpp408 const float* batch_start = lstm_input[b].data() + i * n_input; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/
H A 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
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dindirection.c226 size_t batch_start) { in pytorch_qnnp_indirection_init_dwconv()
410 size_t batch_start) { in pytorch_qnnp_indirection_init_maxpool2d()
/aosp_15_r20/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
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/
Dsvdf.mod.py132 batch_start = i * input_size * batches variable
Dsvdf_bias_present.mod.py132 batch_start = i * input_size * batches variable
Dsvdf2.mod.py147 batch_start = i * input_size * batches variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Dsvdf_bias_present_float16.mod.py132 batch_start = i * input_size * batches variable
Dsvdf_float16.mod.py132 batch_start = i * input_size * batches variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_1/
Dsvdf_relaxed.mod.py133 batch_start = i * input_size * batches variable
Dsvdf_bias_present_relaxed.mod.py133 batch_start = i * input_size * batches variable
Dsvdf2_relaxed.mod.py148 batch_start = i * input_size * batches variable
/aosp_15_r20/external/mesa3d/src/intel/executor/
H A Dexecutor.h44 uint64_t batch_start; member
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DEmbedding.cu65 for(int batch_start = 0; batch_start < n; batch_start += blockDim.x*blockDim.y) in embedding_backward_feature_kernel() local
/aosp_15_r20/external/XNNPACK/src/
H A Doperator-run.c937 size_t batch_start, in xnn_compute_prelu()
1159 size_t batch_start, in xnn_compute_vmulcaddc()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Ddepthwise_conv_hybrid.h146 int batch_start = 0; in DepthwiseConvHybridGeneral() local

12