Home
last modified time | relevance | path

Searched defs:DepthwiseArgs (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddepthwise_conv_op.h25 struct DepthwiseArgs { struct
43 DepthwiseArgs() in DepthwiseArgs() argument
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/assembly/
H A Ddepthwise.hpp45 struct DepthwiseArgs struct
47 const CPUInfo *cpu_info;
49 unsigned int kernel_rows, kernel_cols;
50 unsigned int stride_rows, stride_cols;
52 unsigned int n_batches, input_rows, input_cols, input_channels;
53 unsigned int output_rows, output_cols;
54 unsigned int channel_multiplier;
56 PaddingValues padding;
58 arm_gemm::Activation activation;
60 const DepthwiseConfig *config;
[all …]
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/
H A Ddepthwise_implementation_constraints.hpp140 bool qp_has_no_left_shift(const DepthwiseArgs &, const void *_qp) in qp_has_no_left_shift()
149 bool qp_zero_a_offset(const DepthwiseArgs &, const void *_qp) in qp_zero_a_offset()
156 template <typename T> bool qp_skip_clamp(const DepthwiseArgs &, const void *_qp) in qp_skip_clamp()
H A Ddepthwise_fp16.cpp75 unsigned int not_preferred(const DepthwiseArgs &, const Nothing &) in not_preferred()
H A Ddepthwise_s8q.cpp73 bool qp_weights_are_symmetric(const DepthwiseArgs &, const void *_qp) in qp_weights_are_symmetric()
H A Ddepthwise_fp32.cpp107 unsigned int not_preferred(const DepthwiseArgs &, const Nothing &) in not_preferred()