1 | /******************************************************************************* |
2 | * Copyright 2017-2018 Intel Corporation |
3 | * |
4 | * Licensed under the Apache License, Version 2.0 (the "License"); |
5 | * you may not use this file except in compliance with the License. |
6 | * You may obtain a copy of the License at |
7 | * |
8 | * http://www.apache.org/licenses/LICENSE-2.0 |
9 | * |
10 | * Unless required by applicable law or agreed to in writing, software |
11 | * distributed under the License is distributed on an "AS IS" BASIS, |
12 | * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
13 | * See the License for the specific language governing permissions and |
14 | * limitations under the License. |
15 | *******************************************************************************/ |
16 | |
17 | #include "cpu_engine.hpp" |
18 | |
19 | /* |
20 | #include "cpu/ref_concat.hpp" |
21 | #include "cpu/simple_concat.hpp" |
22 | */ |
23 | |
24 | namespace mkldnn { |
25 | namespace impl { |
26 | namespace cpu { |
27 | |
28 | using cpd_create_f = mkldnn::impl::engine_t::concat_primitive_desc_create_f; |
29 | |
30 | namespace { |
31 | #define INSTANCE(...) __VA_ARGS__::pd_t::create |
32 | static const cpd_create_f cpu_concat_impl_list[] = { |
33 | /* |
34 | INSTANCE(simple_concat_t<data_type::f32>), |
35 | INSTANCE(simple_concat_t<data_type::u8>), |
36 | INSTANCE(simple_concat_t<data_type::s8>), |
37 | INSTANCE(simple_concat_t<data_type::s32>), |
38 | INSTANCE(ref_concat_t), |
39 | */ |
40 | nullptr, |
41 | }; |
42 | #undef INSTANCE |
43 | } |
44 | |
45 | const cpd_create_f *cpu_engine_t::get_concat_implementation_list() const { |
46 | return cpu_concat_impl_list; |
47 | } |
48 | |
49 | } |
50 | } |
51 | } |
52 | |