Searched refs:beginMaskTensor (Results 1 – 1 of 1) sorted by relevance
57 std::shared_ptr<NNTensor> beginMaskTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveBeginMaskTensor() local60 beginMaskTensor->SetBuffer(beginMaskValue, sizeof(int64_t)); in SaveBeginMaskTensor()61 m_allTensors.emplace_back(beginMaskTensor); in SaveBeginMaskTensor()371 std::shared_ptr<NNTensor> beginMaskTensor = TransToNNTensor(OH_NN_INT64, paramDim, nullptr, variable373 beginMaskTensor->SetBuffer(nullptr, 0);374 m_allTensors.emplace_back(beginMaskTensor);