1/**
2 * Copyright (c) Glow Contributors. See CONTRIBUTORS file.
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#ifdef GLOW_WITH_CPU
17
18BB.newBackendSpecificNode("CPUMaxSplat")
19 .addInput("Input")
20 .addResult("Input.getType()")
21 .addMember(MemberType::Float, "SplatValue")
22 .setDocstring("A Max node with one splat input; CPU specific.");
23
24BB.newBackendSpecificNode("CPUConvDKKC8")
25 .addInput("Input")
26 .addInput("Filter")
27 .addInput("Bias")
28 .addMember(MemberType::VectorUnsigned, "Kernels")
29 .addMember(MemberType::VectorUnsigned, "Strides")
30 .addMember(MemberType::VectorUnsigned, "Pads")
31 .addMember(MemberType::Unsigned, "Group")
32 .addResultFromCtorArg()
33 .setDocstring("This is a cpu-specific convolution implementation where the "
34 "filter is transposed to the shape [D/8, K, K, C, 8]");
35
36BB.includeBackendSpecificVerification("glow/CPUSpecificNodesVerification.h");
37
38#endif // GLOW_WITH_CPU
39