depthwise_implementation.hpp File Reference

#include "depthwise.hpp"
#include <cstddef>
#include <functional>

Go to the source code of this file.

Data Structures
struct	DepthwiseImplementation< TInput, TWeight, TOutput, OutputStage >

Namespaces
	arm_conv
	arm_conv::depthwise

Functions
template<typename TInput , typename TWeight = TInput, typename TOutput = TInput, class OutputStage = Nothing>
const DepthwiseImplementation< TInput, TWeight, TOutput, OutputStage > *	depthwise_implementation_list ()
template<typename TInput , typename TWeight = TInput, typename TOutput = TInput, class OutputStage = Nothing>
bool	find_implementation (const DepthwiseArgs &args, const OutputStage &os, const DepthwiseImplementation< TInput, TWeight, TOutput, OutputStage > *&selected)
template<typename TInput , typename TWeight , typename TOutput , class OutputStage >
std::vector< KernelDescription >	get_compatible_kernels (const DepthwiseArgs &args, const OutputStage &os)
template<typename TInput , typename TWeight , typename TOutput , class OutputStage >
UniqueDepthwiseCommon< TInput, TWeight, TOutput >	depthwise (const DepthwiseArgs &args, const OutputStage &os)