Searched refs:output_activation_max (Results 1 - 10 of 10) sorted by relevance

/frameworks/ml/nn/common/operations/
H A DPooling.cpp66 int32_t output_activation_max = 0; local
70 &output_activation_max);
77 output_activation_min, output_activation_max, \
140 int32_t output_activation_max = 0; local
144 &output_activation_max);
151 output_activation_min, output_activation_max, \
H A DFullyConnected.cpp56 int32_t output_activation_max = 0; local
66 &output_activation_max);
78 output_activation_min, output_activation_max, \
H A DSimpleMath.cpp93 int32_t output_activation_max; local
96 &output_activation_max);
106 output_activation_min, output_activation_max, \
117 output_activation_min, output_activation_max, \
176 int32_t output_activation_max; local
179 &output_activation_max);
187 output_activation_min, output_activation_max, \
H A DDepthwiseConv2D.cpp79 int32_t output_activation_max = 0; local
90 &output_activation_max);
103 output_activation_min, output_activation_max, \
H A DConv2D.cpp111 int32_t output_activation_max = 0; local
121 &output_activation_max);
134 output_activation_min, output_activation_max, \
H A DActivation.cpp96 int32_t output_activation_max = 0; \
100 &output_activation_max); \
103 *outputData = std::min((uint8_t)output_activation_max, \
/frameworks/ml/nn/common/operations/internal/reference/
H A Ddepthwiseconv_uint8.h38 int32 output_activation_max, uint8* output_data,
45 DCHECK_LE(output_activation_min, output_activation_max);
48 DCHECK_EQ(output_activation_max, 255);
94 acc = std::min(acc, output_activation_max);
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 Dreference_ops.h225 int32 output_activation_min, int32 output_activation_max,
236 DCHECK_LE(output_activation_min, output_activation_max);
239 DCHECK_EQ(output_activation_max, 255);
285 acc = std::min(acc, output_activation_max);
416 int32 output_activation_max, uint8* output_data,
423 DCHECK_LE(output_activation_min, output_activation_max);
426 DCHECK_EQ(output_activation_max, 255);
454 acc = std::min(acc, output_activation_max);
720 int32 output_activation_min, int32 output_activation_max,
727 DCHECK_LE(output_activation_min, output_activation_max);
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 Doptimized_ops.h375 int32 output_activation_max, uint8* output_data,
519 res8 = vmin_u8(res8, vdup_n_u8(output_activation_max));
541 int32 output_activation_max) {
552 clamp_stage.max = output_activation_max;
571 int32 output_activation_max) {
573 DCHECK_EQ(output_activation_max, 255);
595 int32 output_activation_max, uint8* output_data,
617 output_shift, output_activation_min, output_activation_max, output_data,
640 output_activation_min, output_activation_max);
814 int32 output_activation_min, int32 output_activation_max,
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...]
H A Ddepthwiseconv_uint8.h1353 int32 output_activation_max, uint8* output_data,
1361 DCHECK_LE(output_activation_min, output_activation_max);
1364 DCHECK_EQ(output_activation_max, 255);
1482 vdupq_n_s32(output_activation_max);
1593 acc = std::min(acc, output_activation_max);
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

Completed in 110 milliseconds