Home
last modified time | relevance | path

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

/ohos5.0/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, variable
461 hiddenSizeTensor->SetBuffer(hiddenSizeValue, sizeof(float));
462 m_allTensors.emplace_back(hiddenSizeTensor);
1023 std::shared_ptr<NNTensor> hiddenSizeTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
1025 m_allTensors.emplace_back(hiddenSizeTensor);