24.02
DepthToSpace.hpp
Go to the documentation of this file.
1
//
2
// Copyright © 2019 Arm Ltd. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#pragma once
7
8
#include <
armnn/Descriptors.hpp
>
9
#include <
armnn/Tensor.hpp
>
10
11
namespace
armnn
12
{
13
14
void
DepthToSpace
(
const
TensorInfo& inputInfo,
15
const
DepthToSpaceDescriptor
& descriptor,
16
const
void
* inputData,
17
void
* outputData,
18
unsigned
int
dataTypeSize);
19
20
}
// namespace armnn
armnn::DepthToSpaceDescriptor
SpaceToDepthDescriptor DepthToSpaceDescriptor
A DepthToSpaceDescriptor for the DepthToSpaceLayer.
Definition:
Descriptors.hpp:1099
Descriptors.hpp
Tensor.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
armnn::DepthToSpace
void DepthToSpace(const TensorInfo &inputInfo, const DepthToSpaceDescriptor &descriptor, const void *inputData, void *outputData, unsigned int dataTypeSize)
Definition:
DepthToSpace.cpp:18
src
backends
reference
workloads
DepthToSpace.hpp
Generated on Wed Feb 14 2024 16:36:17 for Arm NN by
1.8.17