Set function for data buffer of user memory primitive.
| 1386 | |
| 1387 | /// Set function for data buffer of user memory primitive. |
| 1388 | inline void SetUsrMemDataHandle(void* data_buffer, |
| 1389 | std::shared_ptr<stream> t_stream = nullptr) { |
| 1390 | CHECK_NOTNULL(user_memory_); |
| 1391 | CHECK_NOTNULL(data_buffer); |
| 1392 | #ifdef ENABLE_DNNL_THREADPOOL |
| 1393 | user_memory_->set_data_handle(data_buffer, *t_stream); |
| 1394 | #else |
| 1395 | user_memory_->set_data_handle(data_buffer); |
| 1396 | #endif // ENABLE_DNNL_THREADPOOL |
| 1397 | } |
| 1398 | |
| 1399 | /// Set function for data buffer of user memory primitive. |
| 1400 | inline void SetUsrMemDataHandle(const Tensor* tensor, |