Searched refs:SaveDropout (Results 1 - 1 of 1) sorted by relevance
| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | lstm_test.cpp | 45 void SaveDropout(OH_NN_DataType dataType,
134 void LSTMBuilderTest::SaveDropout(OH_NN_DataType dataType,
in SaveDropout() function in OHOS::NeuralNetworkRuntime::UnitTest::LSTMBuilderTest
238 SaveDropout(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_DROPOUT);
in HWTEST_F()
263 SaveDropout(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_DROPOUT);
in HWTEST_F()
293 SaveDropout(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_DROPOUT);
in HWTEST_F()
321 SaveDropout(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_DROPOUT);
in HWTEST_F()
375 SaveDropout(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_DROPOUT);
in HWTEST_F()
405 SaveDropout(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_DROPOUT);
in HWTEST_F()
435 SaveDropout(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_DROPOUT);
in HWTEST_F()
465 SaveDropout(OH_NN_FLOAT3 in HWTEST_F()
[all...] |
Completed in 3 milliseconds