Searched refs:SaveZoneoutHidden (Results 1 - 1 of 1) sorted by relevance
| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | lstm_test.cpp | 49 void SaveZoneoutHidden(OH_NN_DataType dataType,
154 void LSTMBuilderTest::SaveZoneoutHidden(OH_NN_DataType dataType,
in SaveZoneoutHidden() function in OHOS::NeuralNetworkRuntime::UnitTest::LSTMBuilderTest
240 SaveZoneoutHidden(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_ZONEOUT_HIDDEN);
in HWTEST_F()
265 SaveZoneoutHidden(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_ZONEOUT_HIDDEN);
in HWTEST_F()
295 SaveZoneoutHidden(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_ZONEOUT_HIDDEN);
in HWTEST_F()
323 SaveZoneoutHidden(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_ZONEOUT_HIDDEN);
in HWTEST_F()
377 SaveZoneoutHidden(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_ZONEOUT_HIDDEN);
in HWTEST_F()
407 SaveZoneoutHidden(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_ZONEOUT_HIDDEN);
in HWTEST_F()
437 SaveZoneoutHidden(OH_NN_FLOAT32, m_paramDim, nullptr, OH_NN_LSTM_ZONEOUT_HIDDEN);
in HWTEST_F()
467 SaveZoneoutHidden(OH_NN_FLOAT3 in HWTEST_F()
[all...] |
Completed in 2 milliseconds