ArmNN
NotReleased
PermuteAndBatchToSpaceAsDepthToSpace.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2019 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
#pragma once
6
7
#include "
Optimization.hpp
"
8
9
namespace
armnn
10
{
11
namespace
optimizations
12
{
13
17
class
PermuteAndBatchToSpaceAsDepthToSpaceImpl
18
{
19
public
:
20
void
Run
(
Graph
& graph,
InputSlot
& connection)
const
;
21
};
22
23
using
PermuteAndBatchToSpaceAsDepthToSpace
=
24
OptimizeForConnection<PermuteLayer, BatchToSpaceNdLayer, PermuteAndBatchToSpaceAsDepthToSpaceImpl>
;
25
26
}
// namespace optimizations
27
}
// namespace armnn
armnn::InputSlot
Definition:
Layer.hpp:41
armnn::optimizations::PermuteAndBatchToSpaceAsDepthToSpaceImpl
Definition:
PermuteAndBatchToSpaceAsDepthToSpace.hpp:17
Optimization.hpp
armnn::OptimizeForConnection
Definition:
Optimization.hpp:116
armnn
Definition:
BackendHelper.hpp:11
armnn::optimizations::PermuteAndBatchToSpaceAsDepthToSpaceImpl::Run
void Run(Graph &graph, InputSlot &connection) const
Definition:
PermuteAndBatchToSpaceAsDepthToSpace.cpp:11
armnn::Graph
Definition:
Graph.hpp:29
src
armnn
optimizations
PermuteAndBatchToSpaceAsDepthToSpace.hpp
Generated on Fri Mar 13 2020 16:06:54 for ArmNN by
1.8.13