|
|
@ -125,7 +125,6 @@ class CUDNNDeviceContext : public CUDADeviceContext {
|
|
|
|
class MKLDNNDeviceContext : public CPUDeviceContext {
|
|
|
|
class MKLDNNDeviceContext : public CPUDeviceContext {
|
|
|
|
public:
|
|
|
|
public:
|
|
|
|
explicit MKLDNNDeviceContext(CPUPlace place);
|
|
|
|
explicit MKLDNNDeviceContext(CPUPlace place);
|
|
|
|
virtual ~MKLDNNDeviceContext();
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
/* \brief Add new element: memory, primitive or primitive desc */
|
|
|
|
/* \brief Add new element: memory, primitive or primitive desc */
|
|
|
|
template <typename T>
|
|
|
|
template <typename T>
|
|
|
|