| /frameworks/ml/nn/common/operations/ |
| H A D | FullyConnected.cpp | 55 int32_t output_activation_min = 0; local
65 &output_activation_min,
78 output_activation_min, output_activation_max, \
|
| H A D | Pooling.cpp | 65 int32_t output_activation_min = 0; local
69 &output_activation_min,
77 output_activation_min, output_activation_max, \
139 int32_t output_activation_min = 0; local
143 &output_activation_min,
151 output_activation_min, output_activation_max, \
|
| H A D | Conv2D.cpp | 110 int32_t output_activation_min = 0; local
120 &output_activation_min,
134 output_activation_min, output_activation_max, \
|
| H A D | DepthwiseConv2D.cpp | 78 int32_t output_activation_min = 0; local
89 &output_activation_min,
103 output_activation_min, output_activation_max, \
|
| H A D | SimpleMath.cpp | 92 int32_t output_activation_min; local
95 &output_activation_min,
106 output_activation_min, output_activation_max, \
117 output_activation_min, output_activation_max, \
175 int32_t output_activation_min; local
178 &output_activation_min,
187 output_activation_min, output_activation_max, \
|
| /frameworks/ml/nn/common/operations/internal/reference/ |
| H A D | depthwiseconv_uint8.h | 37 int output_shift, int32 output_activation_min,
45 DCHECK_LE(output_activation_min, output_activation_max);
47 DCHECK_EQ(output_activation_min, 0);
93 acc = std::max(acc, output_activation_min);
30 DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, int32 input_offset, const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, const int32* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, int depth_multiplier, int32 output_offset, int32 output_multiplier, int output_shift, int32 output_activation_min, int32 output_activation_max, uint8* output_data, const Dims<4>& output_dims) argument
|
| H A D | reference_ops.h | 225 int32 output_activation_min, int32 output_activation_max,
236 DCHECK_LE(output_activation_min, output_activation_max);
238 DCHECK_EQ(output_activation_min, 0);
284 acc = std::max(acc, output_activation_min);
415 int output_shift, int32 output_activation_min,
423 DCHECK_LE(output_activation_min, output_activation_max);
425 DCHECK_EQ(output_activation_min, 0);
453 acc = std::max(acc, output_activation_min);
720 int32 output_activation_min, int32 output_activation_max,
727 DCHECK_LE(output_activation_min, output_activation_ma
219 Conv(const uint8* input_data, const Dims<4>& input_dims, int32 input_offset, const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, const int32* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, int32 output_offset, int32 output_multiplier, int output_shift, int32 output_activation_min, int32 output_activation_max, uint8* output_data, const Dims<4>& output_dims, uint8* im2col_data, const Dims<4>& im2col_dims, gemmlowp::GemmContext* gemm_context) argument
410 FullyConnected(const uint8* input_data, const Dims<4>& input_dims, int32 input_offset, const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, const int32* bias_data, const Dims<4>& bias_dims, int32 output_offset, int32 output_multiplier, int output_shift, int32 output_activation_min, int32 output_activation_max, uint8* output_data, const Dims<4>& output_dims, gemmlowp::GemmContext*) argument
[all...] |
| /frameworks/ml/nn/common/operations/internal/optimized/ |
| H A D | depthwiseconv_uint8.h | 1352 int output_shift, int32 output_activation_min,
1361 DCHECK_LE(output_activation_min, output_activation_max);
1363 DCHECK_EQ(output_activation_min, 0);
1480 vdupq_n_s32(output_activation_min);
1592 acc = std::max(acc, output_activation_min);
1345 DepthwiseConv(const uint8* input_data, const Dims<4>& input_dims, int32 input_offset, const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, const int32* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, int depth_multiplier, int32 output_offset, int32 output_multiplier, int output_shift, int32 output_activation_min, int32 output_activation_max, uint8* output_data, const Dims<4>& output_dims) argument
|
| H A D | optimized_ops.h | 374 int output_shift, int32 output_activation_min,
518 res8 = vmax_u8(res8, vdup_n_u8(output_activation_min));
540 int output_shift, int32 output_activation_min,
551 clamp_stage.min = output_activation_min;
570 int output_shift, int32 output_activation_min,
572 DCHECK_EQ(output_activation_min, 0);
594 int output_shift, int32 output_activation_min,
617 output_shift, output_activation_min, output_activation_max, output_data,
640 output_activation_min, output_activation_max);
814 int32 output_activation_min, int3
369 FullyConnectedAsGEMV(const uint8* input_data, const Dims<4>& input_dims, int32 input_offset, const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, const int32* bias_data, const Dims<4>& bias_dims, int32 output_offset, int32 output_multiplier, int output_shift, int32 output_activation_min, int32 output_activation_max, uint8* output_data, const Dims<4>& output_dims) argument
538 Make(const int32* bias_data, int output_rows, int32 output_offset, int32 output_multiplier, int output_shift, int32 output_activation_min, int32 output_activation_max) argument
568 Make(const int32* bias_data, int output_rows, int32 output_offset, int32 output_multiplier, int output_shift, int32 output_activation_min, int32 output_activation_max) argument
589 FullyConnected(const uint8* input_data, const Dims<4>& input_dims, int32 input_offset, const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, const int32* bias_data, const Dims<4>& bias_dims, int32 output_offset, int32 output_multiplier, int output_shift, int32 output_activation_min, int32 output_activation_max, uint8* output_data, const Dims<4>& output_dims, gemmlowp::GemmContext* gemm_context) argument
808 Conv(const uint8* input_data, const Dims<4>& input_dims, int32 input_offset, const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, const int32* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, int32 output_offset, int32 output_multiplier, int output_shift, int32 output_activation_min, int32 output_activation_max, uint8* output_data, const Dims<4>& output_dims, uint8* im2col_data, const Dims<4>& im2col_dims, gemmlowp::GemmContext* gemm_context) argument
947 ConvAsGemm(const uint8* input_data, const Dims<4>& input_dims, int32 input_offset, const uint8* filter_data, const Dims<4>& filter_dims, int32 filter_offset, const int32* bias_data, const Dims<4>& bias_dims, int32 output_offset, int32 output_multiplier, int output_shift, int32 output_activation_min, int32 output_activation_max, uint8* output_data, const Dims<4>& output_dims, gemmlowp::GemmContext* gemm_context) argument
[all...] |