/frameworks/ml/nn/common/operations/ |
H A D | FullyConnected.cpp | 56 int32_t output_activation_max = 0; local 66 &output_activation_max); 78 output_activation_min, output_activation_max, \
|
H A D | Pooling.cpp | 66 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 D | Conv2D.cpp | 111 int32_t output_activation_max = 0; local 121 &output_activation_max); 134 output_activation_min, output_activation_max, \
|
H A D | DepthwiseConv2D.cpp | 79 int32_t output_activation_max = 0; local 90 &output_activation_max); 103 output_activation_min, output_activation_max, \
|
H A D | SimpleMath.cpp | 93 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, \
|
/frameworks/ml/nn/common/operations/internal/reference/ |
H A D | depthwiseconv_uint8.h | 38 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 D | reference_ops.h | 225 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 D | depthwiseconv_uint8.h | 1353 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
|
H A D | optimized_ops.h | 375 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...] |