| 228 | } |
| 229 | |
| 230 | inline dnnl::stream* CreateStream(MklDnnThreadPool* eigen_tp, |
| 231 | const engine& engine) { |
| 232 | #ifdef ENABLE_DNNL_THREADPOOL |
| 233 | if (eigen_tp != nullptr) { |
| 234 | stream* tp_stream = |
| 235 | new stream(dnnl::threadpool_interop::make_stream(engine, eigen_tp)); |
| 236 | return tp_stream; |
| 237 | } else { |
| 238 | stream* tp_stream = new stream(engine); |
| 239 | return tp_stream; |
| 240 | } |
| 241 | #else |
| 242 | stream* tp_stream = new stream(engine); |
| 243 | return tp_stream; |
| 244 | #endif // ENABLE_DNNL_THREADPOOL |
| 245 | } |
| 246 | |
| 247 | class MklDnnShape { |
| 248 | private: |
no outgoing calls
no test coverage detected