Searched refs:numClassesTensor (Results 1 – 1 of 1) sorted by relevance
146 std::shared_ptr<NNTensor> numClassesTensor = TransToNNTensor(dataType, dim, quantParam, type); in SetNumClasses() local149 numClassesTensor->SetBuffer(numClassesValue, sizeof(int64_t)); in SetNumClasses()150 m_allTensors.emplace_back(numClassesTensor); in SetNumClasses()539 std::shared_ptr<NNTensor> numClassesTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable542 numClassesTensor->SetBuffer(numClassesValue, sizeof(float));543 m_allTensors.emplace_back(numClassesTensor);1066 std::shared_ptr<NNTensor> numClassesTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable1068 m_allTensors.emplace_back(numClassesTensor);