\brief Returns the device buffer corresponding to tensorName. Returns nullptr if no such tensor can be found.
| 293 | //! Returns nullptr if no such tensor can be found. |
| 294 | //! |
| 295 | void* getDeviceBuffer(const std::string& tensorName) const |
| 296 | { |
| 297 | return getBuffer(false, tensorName); |
| 298 | } |
| 299 | |
| 300 | //! |
| 301 | //! \brief Returns the host buffer corresponding to tensorName. |