Home
last modified time | relevance | path

Searched refs:batch1_start (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dbidirectional_sequence_lstm_test.cc1887 float* batch1_start = lstm_input[1] + i * lstm.num_inputs(); in TEST() local
1888 float* batch1_end = batch1_start + lstm.num_inputs(); in TEST()
1889 lstm.SetInput((2 * i + 1) * lstm.num_inputs(), batch1_start, batch1_end); in TEST()
2590 float* batch1_start = lstm_input[1]; in TEST() local
2591 float* batch1_end = batch1_start + input_sequence_size; in TEST()
2592 lstm.SetInput(input_sequence_size, batch1_start, batch1_end); in TEST()
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTMTest.cpp991 float* batch1_start = lstm_input[1] + i * lstm.num_inputs(); in TEST() local
992 float* batch1_end = batch1_start + lstm.num_inputs(); in TEST()
993 lstm.SetInput(lstm.num_inputs(), batch1_start, batch1_end); in TEST()