Searched defs:DepthwiseArgs (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_op.h | 25 struct DepthwiseArgs { struct
43 DepthwiseArgs() in DepthwiseArgs() argument
|
| /external/ComputeLibrary/src/core/NEON/kernels/assembly/ |
| D | depthwise.hpp | 45 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 …]
|
| /external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/ |
| D | depthwise_implementation_constraints.hpp | 140 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()
|
| D | depthwise_fp16.cpp | 75 unsigned int not_preferred(const DepthwiseArgs &, const Nothing &) in not_preferred()
|
| D | depthwise_s8q.cpp | 73 bool qp_weights_are_symmetric(const DepthwiseArgs &, const void *_qp) in qp_weights_are_symmetric()
|
| D | depthwise_fp32.cpp | 107 unsigned int not_preferred(const DepthwiseArgs &, const Nothing &) in not_preferred()
|