Searched refs:dropoutTensor (Results 1 – 1 of 1) sorted by relevance
137 std::shared_ptr<NNTensor> dropoutTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveDropout() local140 dropoutTensor->SetBuffer(dropoutValue, sizeof(float)); in SaveDropout()141 m_allTensors.emplace_back(dropoutTensor); in SaveDropout()550 std::shared_ptr<NNTensor> dropoutTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable554 dropoutTensor->SetBuffer(dropoutValue, sizeof(int64_t));555 m_allTensors.emplace_back(dropoutTensor);1107 std::shared_ptr<NNTensor> dropoutTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable1109 m_allTensors.emplace_back(dropoutTensor);