Searched refs:numLayersTensor (Results 1 – 1 of 1) sorted by relevance
117 std::shared_ptr<NNTensor> numLayersTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveNumLayers() local120 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, variable492 numLayersTensor->SetBuffer(numLayersValue, sizeof(float));493 m_allTensors.emplace_back(numLayersTensor);1051 std::shared_ptr<NNTensor> numLayersTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable1053 m_allTensors.emplace_back(numLayersTensor);