cross_channel_norm
tensor-tang 7 years ago
parent 31fda46cf4
commit b0ba2b061e

@ -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>

@ -14,7 +14,7 @@ limitations under the License. */
#pragma once #pragma once
#include "mkldnn.hpp" #include <mkldnn.hpp>
namespace paddle { namespace paddle {
namespace platform { namespace platform {

Loading…
Cancel
Save