summaryrefslogtreecommitdiffstats
path: root/thirdparty/oidn/mkl-dnn/src/cpu/ref_concat.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'thirdparty/oidn/mkl-dnn/src/cpu/ref_concat.hpp')
-rw-r--r--thirdparty/oidn/mkl-dnn/src/cpu/ref_concat.hpp97
1 files changed, 0 insertions, 97 deletions
diff --git a/thirdparty/oidn/mkl-dnn/src/cpu/ref_concat.hpp b/thirdparty/oidn/mkl-dnn/src/cpu/ref_concat.hpp
deleted file mode 100644
index 4c534b5508..0000000000
--- a/thirdparty/oidn/mkl-dnn/src/cpu/ref_concat.hpp
+++ /dev/null
@@ -1,97 +0,0 @@
-/*******************************************************************************
-* Copyright 2017-2018 Intel Corporation
-*
-* Licensed under the Apache License, Version 2.0 (the "License");
-* you may not use this file except in compliance with the License.
-* You may obtain a copy of the License at
-*
-* http://www.apache.org/licenses/LICENSE-2.0
-*
-* Unless required by applicable law or agreed to in writing, software
-* distributed under the License is distributed on an "AS IS" BASIS,
-* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-* See the License for the specific language governing permissions and
-* limitations under the License.
-*******************************************************************************/
-
-#ifndef REF_CONCAT_HPP
-#define REF_CONCAT_HPP
-
-#include "reorder_pd.hpp"
-
-#include "cpu_concat_pd.hpp"
-#include "cpu_primitive.hpp"
-
-namespace mkldnn {
-namespace impl {
-namespace cpu {
-
-struct ref_concat_t: public cpu_primitive_t {
- struct pd_t: public cpu_concat_pd_t {
- using cpu_concat_pd_t::cpu_concat_pd_t;
-
- pd_t(const pd_t &rhs): cpu_concat_pd_t(rhs) {
- for (size_t i = 0; i < rhs.reorder_pds_.size(); ++i)
- reorder_pds_.push_back(
- (const reorder_pd_t *)rhs.reorder_pds_[i]->clone());
- }
- ~pd_t() { for (auto &rpd: reorder_pds_) delete rpd; }
-
- DECLARE_CONCAT_PD_T("ref:any", ref_concat_t);
-
- status_t init() {
- bool ok = cpu_concat_pd_t::init() == status::success;
- if (!ok) return status::unimplemented;
-
- for (int i = 0; i < n_; ++i) {
- auto r_impls = engine_->get_reorder_implementation_list();
- for (auto r = r_impls; *r; ++r) {
- const primitive_attr_t attr; /* alpha == 1. */
- reorder_pd_t *r_pd = nullptr;
- if ((*r)(&r_pd, engine_, &attr, engine_, src_md(i),
- engine_, src_image_md(i)) == status::success) {
- r_pd->init_info();
- reorder_pds_.push_back(r_pd);
- break;
- }
- }
- }
-
- ok = reorder_pds_.size() == (size_t)n_;
- return ok ? status::success : status::unimplemented;
- }
-
- nstl::vector<const reorder_pd_t *> reorder_pds_;
- };
-
- ref_concat_t(const pd_t *apd): cpu_primitive_t(apd) {
- const int n = pd()->n_inputs();
- reorders_.resize(n);
- for (int i = 0; i < n; ++i)
- pd()->reorder_pds_[i]->create_primitive(&reorders_[i]);
- }
-
- ~ref_concat_t() { for (auto &r: reorders_) delete r; }
-
- virtual status_t execute(const exec_ctx_t &ctx) const override {
- const auto n = pd()->n_inputs();
- for (int i = 0; i < n; ++i) {
- exec_args_t r_args;
- r_args[MKLDNN_ARG_SRC] = ctx.args().at(MKLDNN_ARG_MULTIPLE_SRC + i);
- r_args[MKLDNN_ARG_DST] = ctx.args().at(MKLDNN_ARG_DST);
- exec_ctx_t r_ctx(ctx.stream(), std::move(r_args));
- reorders_[i]->execute(r_ctx);
- }
- return status::success;
- }
-
-private:
- const pd_t *pd() const { return (const pd_t *)primitive_t::pd(); }
- nstl::vector<primitive_t *> reorders_;
-};
-
-}
-}
-}
-
-#endif