/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
H A D | cblas_conv.h | 38 int pad_height, float output_activation_min, 53 pad_width, pad_height, filter_height, filter_width, 0, 34 Conv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, float output_activation_min, float output_activation_max, float* output_data, const Dims<4>& output_dims, float* im2col_data, const Dims<4>& im2col_dims) argument
|
H A D | multithreaded_conv.h | 120 int pad_width, int pad_height, TfLitePadding padding, 138 pad_width == 0 && pad_height == 0) { 168 int pad_height, TfLitePadding padding, 184 output_depth, stride_height, stride_width, pad_height, pad_width, 116 operator ()(const T* input_data, T* im2col_buffer, int input_batches, int input_height, int input_width, int input_depth, const T* filter_data, int filter_height, int filter_width, int filter_count, int stride_rows, int stride_cols, int pad_width, int pad_height, TfLitePadding padding, T* output_data, int output_height, int output_width) argument 164 Conv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, TfLitePadding padding, float output_activation_min, float output_activation_max, float* output_data, const Dims<4>& output_dims, float* im2col_data, const Dims<4>& im2col_dims) argument
|
H A D | depthwiseconv_float.h | 914 int pad_height, int depth_multiplier, 997 const int in_y_origin = (out_y * stride_height) - pad_height; 1076 int pad_height, int depth_multiplier, float* output_data, 1081 bias_dims, stride_width, stride_height, pad_width, pad_height, 1091 int pad_width, int pad_height, int depth_multiplier, 1094 bias_dims, stride, stride, pad_width, pad_height, 910 DepthwiseConv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, int depth_multiplier, float output_activation_min, float output_activation_max, float* output_data, const Dims<4>& output_dims) argument 1072 DepthwiseConv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, int depth_multiplier, float* output_data, const Dims<4>& output_dims) argument 1088 DepthwiseConv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride, int pad_width, int pad_height, int depth_multiplier, float* output_data, const Dims<4>& output_dims) argument
|
H A D | depthwiseconv_uint8.h | 1676 int pad_height, int depth_multiplier, 1769 const int in_y_origin = (out_y * stride_height) - pad_height; 1928 int pad_height, int depth_multiplier, int32 output_offset, 1938 stride_height, pad_width, pad_height, depth_multiplier, 1950 int pad_width, int pad_height, int depth_multiplier, 1957 stride, pad_width, pad_height, depth_multiplier, 1671 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 1923 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 1946 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, 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 | 638 int stride_width, int stride_height, int pad_width, int pad_height, 646 const int ih_ungated_start = h * stride_height - pad_height; 723 int stride_height, int pad_width, int pad_height, int kheight, 744 pad_width, pad_height, input_width, input_height, input_depth, 755 int pad_width, int pad_height, int kheight, int kwidth, 757 Im2col(input_data, input_dims, stride, stride, pad_width, pad_height, kheight, 765 int pad_height, float output_activation_min, 782 pad_height, filter_height, filter_width, 0, im2col_data, 813 int stride_height, int pad_width, int pad_height, float* output_data, 819 stride_width, stride_height, pad_width, pad_height, 636 ExtractPatchIntoBufferColumn( const Dims<4>& input_dims, int w, int h, int b, int kheight, int kwidth, int stride_width, int stride_height, int pad_width, int pad_height, int in_width, int in_height, int in_depth, int single_buffer_length, int buffer_id, const T* in_data, T* conv_buffer_data, uint8 byte_zero) argument 722 Im2col(const T* input_data, const Dims<4>& input_dims, int stride_width, int stride_height, int pad_width, int pad_height, int kheight, int kwidth, uint8 byte_zero, T* output_data, const Dims<4>& output_dims) argument 754 Im2col(const T* input_data, const Dims<4>& input_dims, int stride, int pad_width, int pad_height, int kheight, int kwidth, uint8 byte_zero, T* output_data, const Dims<4>& output_dims) argument 761 Conv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, float output_activation_min, float output_activation_max, float* output_data, const Dims<4>& output_dims, float* im2col_data, const Dims<4>& im2col_dims) argument 810 Conv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, float* output_data, const Dims<4>& output_dims, float* im2col_data, const Dims<4>& im2col_dims) argument 826 Conv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride, int pad_width, int pad_height, float* output_data, const Dims<4>& output_dims, float* im2col_data, const Dims<4>& im2col_dims) argument 837 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 910 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 939 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, 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 994 Im2col(const T* input_data, const Dims<4>& input_dims, int stride, int pad_width, int pad_height, int kheight, int kwidth, uint8 byte_zero, T* output_data, const Dims<4>& output_dims) argument [all...] |
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
H A D | depthwiseconv_float.h | 29 int pad_height, int depth_multiplier, 51 const int in_y_origin = (out_y * stride_height) - pad_height; 90 int pad_height, int depth_multiplier, float* output_data, 95 bias_dims, stride_width, stride_height, pad_width, pad_height, 105 int pad_width, int pad_height, int depth_multiplier, 108 bias_dims, stride, stride, pad_width, pad_height, 25 DepthwiseConv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, int depth_multiplier, float output_activation_min, float output_activation_max, float* output_data, const Dims<4>& output_dims) argument 86 DepthwiseConv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, int depth_multiplier, float* output_data, const Dims<4>& output_dims) argument 102 DepthwiseConv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride, int pad_width, int pad_height, int depth_multiplier, float* output_data, const Dims<4>& output_dims) argument
|
H A D | depthwiseconv_uint8.h | 34 int pad_height, int depth_multiplier, 57 const int in_y_origin = (out_y * stride_height) - pad_height; 100 int pad_height, int depth_multiplier, int32 output_offset, 110 stride_height, pad_width, pad_height, depth_multiplier, 122 int pad_width, int pad_height, int depth_multiplier, 129 stride, pad_width, pad_height, depth_multiplier, 29 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 95 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 118 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, 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 | 161 int pad_height, float output_activation_min, 184 const int in_y_origin = (out_y * stride_height) - pad_height; 224 int stride_height, int pad_width, int pad_height, float* output_data, 230 stride_width, stride_height, pad_width, pad_height, 240 int pad_width, int pad_height, float* output_data, 244 bias_dims, stride, stride, pad_width, pad_height, output_data, 253 int pad_height, int32 output_offset, int32 output_multiplier, 278 const int in_y_origin = (out_y * stride_height) - pad_height; 323 int pad_height, int32 output_offset, int32 output_multiplier, 340 pad_width, pad_height, output_offse 157 Conv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, float output_activation_min, float output_activation_max, float* output_data, const Dims<4>& output_dims, float* im2col_data, const Dims<4>& im2col_dims) argument 221 Conv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride_width, int stride_height, int pad_width, int pad_height, float* output_data, const Dims<4>& output_dims, float* im2col_data, const Dims<4>& im2col_dims) argument 237 Conv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride, int pad_width, int pad_height, float* output_data, const Dims<4>& output_dims, float* im2col_data, const Dims<4>& im2col_dims) argument 248 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 318 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 347 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, 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 [all...] |
/external/tensorflow/tensorflow/core/kernels/neon/ |
H A D | depthwiseconv_float.h | 556 int pad_width, int pad_height, int depth_multiplier, 631 const int in_y_origin = (out_y * stride) - pad_height; 553 DepthwiseConv(const float* input_data, const Dims<4>& input_dims, const float* filter_data, const Dims<4>& filter_dims, const float* bias_data, const Dims<4>& bias_dims, int stride, int pad_width, int pad_height, int depth_multiplier, float* output_data, const Dims<4>& output_dims) argument
|