Searched refs:num_outputs (Results 1 – 2 of 2) sorted by relevance
184 uint32_t num_outputs() const { return n_output_; } in num_outputs() function in android::nn::wrapper::LSTMOpModel368 float* golden_start = lstm_golden_output + i * lstm.num_outputs(); in TEST()369 float* golden_end = golden_start + lstm.num_outputs(); in TEST()465 float* golden_start = lstm_golden_output + i * lstm.num_outputs(); in TEST()466 float* golden_end = golden_start + lstm.num_outputs(); in TEST()997 float* golden_start_batch0 = lstm_golden_output[0] + i * lstm.num_outputs(); in TEST()998 float* golden_end_batch0 = golden_start_batch0 + lstm.num_outputs(); in TEST()999 float* golden_start_batch1 = lstm_golden_output[1] + i * lstm.num_outputs(); in TEST()1000 float* golden_end_batch1 = golden_start_batch1 + lstm.num_outputs(); in TEST()
196 uint32_t num_outputs() const { return n_output_; } in num_outputs() function in android::nn::wrapper::LayerNormLSTMOpModel