Searched refs:SaveHiddenSize (Results 1 - 1 of 1) sorted by relevance
| /foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | lstm_test.cpp | 39 void SaveHiddenSize(OH_NN_DataType dataType,
104 void LSTMBuilderTest::SaveHiddenSize(OH_NN_DataType dataType,
in SaveHiddenSize() function in OHOS::NeuralNetworkRuntime::UnitTest::LSTMBuilderTest
235 SaveHiddenSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_HIDDEN_SIZE);
in HWTEST_F()
260 SaveHiddenSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_HIDDEN_SIZE);
in HWTEST_F()
290 SaveHiddenSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_HIDDEN_SIZE);
in HWTEST_F()
318 SaveHiddenSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_HIDDEN_SIZE);
in HWTEST_F()
372 SaveHiddenSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_HIDDEN_SIZE);
in HWTEST_F()
402 SaveHiddenSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_HIDDEN_SIZE);
in HWTEST_F()
432 SaveHiddenSize(OH_NN_INT64, m_paramDim, nullptr, OH_NN_LSTM_HIDDEN_SIZE);
in HWTEST_F()
487 SaveHiddenSize(OH_NN_INT6 in HWTEST_F()
[all...] |
Completed in 2 milliseconds