Gather.cpp File Reference

#include "Gather.hpp"
#include <armnn/backends/WorkloadData.hpp>
#include <fmt/format.h>

Include dependency graph for Gather.cpp:

Go to the source code of this file.

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

Functions
template<typename I, typename O>
void	Gather (const TensorInfo &paramsInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo, Decoder< I > &params, const int32_t *indices, Encoder< O > &output, const int32_t axis_int)
template void	Gather (const TensorInfo &paramsInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo, Decoder< float > &params, const int32_t *indices, Encoder< float > &output, const int32_t axis_int)
template void	Gather (const TensorInfo &paramsInfo, const TensorInfo &indicesInfo, const TensorInfo &outputInfo, Decoder< double_t > &params, const int32_t *indices, Encoder< double_t > &output, const int32_t axis_int)