BatchToSpaceNd.cpp File Reference

#include "BatchToSpaceNd.hpp"
#include <armnnUtils/DataLayoutIndexed.hpp>

Include dependency graph for BatchToSpaceNd.cpp:

Go to the source code of this file.

Namespaces
	armnn
	Copyright (c) 2021 ARM Limited and Contributors.

Functions
unsigned int	Offset (const TensorShape &shape, unsigned int batch, unsigned int height, unsigned int width, unsigned int channels, const DataLayoutIndexed &dataLayout)
void	BatchToSpaceNd (const TensorInfo &inputInfo, const TensorInfo &outputInfo, const BatchToSpaceNdDescriptor &params, Decoder< float > &inputData, Encoder< float > &outputData)