Go to the source code of this file.
|
bool | is_valid_input_region (int32_t base_w, uint32_t base_h, uint32_t w, uint32_t h, const DepthwiseConvolutionRunInfo &run_info, const Size2D &dilation) |
|
template<typename T > |
void | depthwise_loop_multiplier1_fp (const ITensor *src, const ITensor *weights, const ITensor *biases, ITensor *dst, const PadStrideInfo &conv_info, const Size2D &dilation, const Window &window, bool has_biases) |
|
template<typename T > |
void | depthwise_loop_generic_fp (const ITensor *src, const ITensor *weights, const ITensor *biases, ITensor *dst, const PadStrideInfo &conv_info, const Size2D &dilation, unsigned int depth_multiplier, const Window &window, bool has_biases) |
|
template<typename T , typename TW > |
void | run_depthwise_float (const ITensor *src, const ITensor *weights, const ITensor *biases, ITensor *dst, const Window &window, bool has_biases, const ConvolutionInfo &info) |
|
template<typename T , typename TW > |
void | run_depthwise_quanitized8bit (const ITensor *src, const ITensor *weights, const ITensor *biases, ITensor *dst, const Window &window, bool has_biases, const ConvolutionInfo &info) |
|
|
constexpr auto | data_layout = DataLayout::NHWC |
|
const size_t | width_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH) |
|
const size_t | height_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT) |
|
const size_t | channel_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::CHANNEL) |
|
constexpr auto | dim_manual_loop = Window::Dimension(0, 0, 0) |
|
constexpr auto | dim_single_unit_step = Window::Dimension(0, 1, 1) |
|
constexpr size_t | vector_size = 8 |
|