Lines Matching refs:num_inputs
184 uint32_t num_inputs() const { return n_input_; } in num_inputs() function in android::nn::wrapper::LSTMOpModel
360 const int input_sequence_size = sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST()
362 float* batch0_start = lstm_input + i * lstm.num_inputs(); in TEST()
363 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST()
457 const int input_sequence_size = sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST()
459 float* batch0_start = lstm_input + i * lstm.num_inputs(); in TEST()
460 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST()
985 const int input_sequence_size = sizeof(lstm_input[0]) / sizeof(float) / (lstm.num_inputs()); in TEST()
987 float* batch0_start = lstm_input[0] + i * lstm.num_inputs(); in TEST()
988 float* batch0_end = batch0_start + lstm.num_inputs(); in TEST()
992 float* batch1_start = lstm_input[1] + i * lstm.num_inputs(); in TEST()
993 float* batch1_end = batch1_start + lstm.num_inputs(); in TEST()
994 lstm.SetInput(lstm.num_inputs(), batch1_start, batch1_end); in TEST()