Searched refs:zoneoutCellTensor (Results 1 – 1 of 1) sorted by relevance
147 std::shared_ptr<NNTensor> zoneoutCellTensor = TransToNNTensor(dataType, dim, quantParam, type); in SaveZoneoutCell() local150 zoneoutCellTensor->SetBuffer(zoneoutCellValue, sizeof(float)); in SaveZoneoutCell()151 m_allTensors.emplace_back(zoneoutCellTensor); in SaveZoneoutCell()581 std::shared_ptr<NNTensor> zoneoutCellTensor = TransToNNTensor(OH_NN_INT64, m_paramDim, variable585 zoneoutCellTensor->SetBuffer(zoneoutCellValue, sizeof(int64_t));586 m_allTensors.emplace_back(zoneoutCellTensor);1135 std::shared_ptr<NNTensor> zoneoutCellTensor = TransToNNTensor(OH_NN_FLOAT32, m_paramDim, variable1137 m_allTensors.emplace_back(zoneoutCellTensor);