summaryrefslogtreecommitdiff
path: root/inference-engine/src/mkldnn_plugin/mkldnn_dims.h
diff options
context:
space:
mode:
Diffstat (limited to 'inference-engine/src/mkldnn_plugin/mkldnn_dims.h')
-rw-r--r--inference-engine/src/mkldnn_plugin/mkldnn_dims.h15
1 files changed, 3 insertions, 12 deletions
diff --git a/inference-engine/src/mkldnn_plugin/mkldnn_dims.h b/inference-engine/src/mkldnn_plugin/mkldnn_dims.h
index 6b0c7b2da..f707f268c 100644
--- a/inference-engine/src/mkldnn_plugin/mkldnn_dims.h
+++ b/inference-engine/src/mkldnn_plugin/mkldnn_dims.h
@@ -9,14 +9,14 @@
#include <vector>
#include <utility>
#include <mkldnn_types.h>
+#include <ie_common.h>
#include <mkldnn.hpp>
namespace MKLDNNPlugin {
class MKLDNNDims {
public:
- MKLDNNDims() {
- }
+ MKLDNNDims() = default;
explicit MKLDNNDims(const InferenceEngine::SizeVector& size) {
dims = std::vector<int>(size.begin(), size.end());
@@ -31,6 +31,7 @@ public:
}
explicit MKLDNNDims(std::initializer_list<int> ilist) : dims(ilist) {}
+ explicit MKLDNNDims(std::initializer_list<size_t > ilist) : dims(ilist.begin(), ilist.end()) {}
InferenceEngine::SizeVector ToSizeVector() const {
InferenceEngine::SizeVector size;
@@ -59,20 +60,10 @@ public:
return size;
}
- void insert(int at, int val) {
- dims.insert(dims.begin() + at, val);
- }
-
void push_back(int val) {
dims.push_back(val);
}
- void swap(int from, int to) {
- int tmp = dims[from];
- dims[from] = dims[to];
- dims[to] = tmp;
- }
-
operator mkldnn::memory::dims() const {
return dims;
}