Home
last modified time | relevance | path

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

/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Dlstm_test.cpp117 std::shared_ptr<NNTensor> numLayersTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveNumLayers() local
120 numLayersTensor->SetBuffer(numLayersValue, sizeof(int64_t)); in SaveNumLayers()
121 m_allTensors.emplace_back(numLayersTensor); in SaveNumLayers()
488 std::shared_ptr<NNTensor> numLayersTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, in HWTEST_F() local
492 numLayersTensor->SetBuffer(numLayersValue, sizeof(float)); in HWTEST_F()
493 m_allTensors.emplace_back(numLayersTensor); in HWTEST_F()
1051 std::shared_ptr<NNTensor> numLayersTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, in HWTEST_F() local
1053 m_allTensors.emplace_back(numLayersTensor); in HWTEST_F()

Completed in 4 milliseconds