Home
last modified time | relevance | path

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

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dlstm_test.cpp157 std::shared_ptr<NNTensor> zoneoutHiddenTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveZoneoutHidden() local
160 zoneoutHiddenTensor->SetBuffer(zoneoutHiddenValue, sizeof(float)); in SaveZoneoutHidden()
161 m_allTensors.emplace_back(zoneoutHiddenTensor); in SaveZoneoutHidden()
612 std::shared_ptr<NNTensor> zoneoutHiddenTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, in HWTEST_F() local
616 zoneoutHiddenTensor->SetBuffer(zoneoutHiddenValue, sizeof(int64_t)); in HWTEST_F()
617 m_allTensors.emplace_back(zoneoutHiddenTensor); in HWTEST_F()
1163 std::shared_ptr<NNTensor> zoneoutHiddenTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, in HWTEST_F() local
1165 m_allTensors.emplace_back(zoneoutHiddenTensor); in HWTEST_F()

Completed in 3 milliseconds