Home
last modified time | relevance | path

Searched defs:batch0_end (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbidirectional_sequence_lstm_test.cc568 float* batch0_end = batch0_start + lstm.num_inputs() * lstm.sequence_length(); in TEST_P() local
738 float* batch0_end = batch0_start + lstm.num_inputs() * lstm.num_batches() * in TEST_P() local
897 float* batch0_end = batch0_start + lstm.num_inputs() * lstm.sequence_length(); in TEST() local
1049 float* batch0_end = batch0_start + lstm.num_inputs() * lstm.sequence_length(); in TEST() local
1200 float* batch0_end = batch0_start + lstm.num_inputs() * lstm.sequence_length(); in TEST() local
1883 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST() local
2587 float* batch0_end = batch0_start + input_sequence_size; in TEST() local
2768 float* batch0_end = batch0_start + lstm.num_inputs() * lstm.sequence_length(); in TEST_P() local
2946 float* batch0_end = batch0_start + lstm.num_inputs() * lstm.sequence_length(); in TEST_P() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTMTest.cpp362 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST() local
459 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST() local
987 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST() local