Searched defs:pad_height (Results 1 - 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
H A Dcblas_conv.h38 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 Dmultithreaded_conv.h120 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 Ddepthwiseconv_float.h914 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 Ddepthwiseconv_uint8.h1676 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 Doptimized_ops.h638 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 Ddepthwiseconv_float.h29 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 Ddepthwiseconv_uint8.h34 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 Dreference_ops.h161 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 Ddepthwiseconv_float.h556 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

Completed in 251 milliseconds