Searched refs:endMaskTensor (Results 1 – 1 of 1) sorted by relevance
68 std::shared_ptr<NNTensor> endMaskTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveEndMaskTensor() local71 endMaskTensor->SetBuffer(endMaskValue, sizeof(int64_t)); in SaveEndMaskTensor()72 m_allTensors.emplace_back(endMaskTensor); in SaveEndMaskTensor()399 std::shared_ptr<NNTensor> endMaskTensor = TransToNNTensor(OH_NN_INT64, paramDim, nullptr, variable401 endMaskTensor->SetBuffer(nullptr, 0);402 m_allTensors.emplace_back(endMaskTensor);