Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dlstm_test.cpp77 … std::shared_ptr<NNTensor> bidirectionalTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveBidirectional() local
80 bidirectionalTensor->SetBuffer(bidirectionalValue, sizeof(bool)); in SaveBidirectional()
81 m_allTensors.emplace_back(bidirectionalTensor); in SaveBidirectional()
364 std::shared_ptr<NNTensor> bidirectionalTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable
368 bidirectionalTensor->SetBuffer(bidirectionalValue, sizeof(int64_t));
369 m_allTensors.emplace_back(bidirectionalTensor);
939 std::shared_ptr<NNTensor> bidirectionalTensor = TransToNNTensor(OH_NN_BOOL, m_paramDim, variable
941 m_allTensors.emplace_back(bidirectionalTensor);