Compute Library
 22.08
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)