1 // Copyright (C) 2018 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
6 * @brief Convinience wrapper class for handling MKL-DNN memory formats.
7 * @file mkldnn_extension_utils.h
14 #include "inference_engine.hpp"
15 #include "mkldnn_memory.h"
17 namespace MKLDNNPlugin {
19 class MKLDNNExtensionUtils {
21 static uint8_t sizeOfDataType(mkldnn::memory::data_type dataType);
22 static mkldnn::memory::data_type IEPrecisionToDataType(InferenceEngine::Precision prec);
23 static InferenceEngine::Precision DataTypeToIEPrecision(mkldnn::memory::data_type dataType);
24 static InferenceEngine::TensorDesc getUninitTensorDesc(const InferenceEngine::TensorDesc& desc);
25 static bool initTensorsAreEqual(InferenceEngine::TensorDesc desc1, InferenceEngine::TensorDesc desc2);
28 } // namespace MKLDNNPlugin