Home
last modified time | relevance | path

Searched refs:hiddenSizeTensor (Results 1 - 1 of 1) sorted by relevance

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dlstm_test.cpp107 std::shared_ptr<NNTensor> hiddenSizeTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveHiddenSize() local
110 hiddenSizeTensor->SetBuffer(hiddenSizeValue, sizeof(int64_t)); in SaveHiddenSize()
111 m_allTensors.emplace_back(hiddenSizeTensor); in SaveHiddenSize()
457 std::shared_ptr<NNTensor> hiddenSizeTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, in HWTEST_F() local
461 hiddenSizeTensor->SetBuffer(hiddenSizeValue, sizeof(float)); in HWTEST_F()
462 m_allTensors.emplace_back(hiddenSizeTensor); in HWTEST_F()
1023 std::shared_ptr<NNTensor> hiddenSizeTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, in HWTEST_F() local
1025 m_allTensors.emplace_back(hiddenSizeTensor); in HWTEST_F()

Completed in 2 milliseconds