Searched defs:stride_width (Results 1 - 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
H A Dcblas_conv.h37 int stride_width, int stride_height, int pad_width,
48 const bool need_im2col = stride_width != 1 || stride_height != 1 ||
52 optimized_ops::Im2col(input_data, input_dims, stride_width, stride_height,
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.h167 int stride_width, int stride_height, int pad_width,
184 output_depth, stride_height, stride_width, pad_height, pad_width,
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.h913 int stride_width, int stride_height, int pad_width,
947 if (!row_accum_func && (stride_width == 1 || ALLOW_STRIDED) && \
1017 row_accum_func(stride_width, input_depth, input_width,
1075 int stride_width, int stride_height, int pad_width,
1081 bias_dims, stride_width, stride_height, 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
H A Ddepthwiseconv_uint8.h1675 int stride_width, int stride_height, int pad_width,
1712 if (!row_accum_func && (stride_width == 1 || ALLOW_STRIDED) && \
1790 stride_width, input_depth, input_width,
1927 int stride_width, int stride_height, int pad_width,
1937 filter_offset, bias_data, bias_dims, stride_width,
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
H A Doptimized_ops.h638 int stride_width, int stride_height, int pad_width, int pad_height,
649 const int iw_ungated_start = w * stride_width - pad_width;
722 void Im2col(const T* input_data, const Dims<4>& input_dims, int stride_width, argument
743 input_dims, w, h, b, kheight, kwidth, stride_width, stride_height,
764 int stride_width, int stride_height, int pad_width,
777 const bool need_im2col = stride_width != 1 || stride_height != 1 ||
781 Im2col(input_data, input_dims, stride_width, stride_height, pad_width,
812 const float* bias_data, const Dims<4>& bias_dims, int stride_width,
819 stride_width, stride_height, pad_width, pad_height,
841 int stride_width, in
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
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
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
[all...]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/
H A Ddepthwiseconv_float.h28 int stride_width, int stride_height, int pad_width,
50 const int in_x_origin = (out_x * stride_width) - pad_width;
89 int stride_width, int stride_height, int pad_width,
95 bias_dims, stride_width, stride_height, 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
H A Ddepthwiseconv_uint8.h33 int stride_width, int stride_height, int pad_width,
56 const int in_x_origin = (out_x * stride_width) - pad_width;
99 int stride_width, int stride_height, int pad_width,
109 filter_offset, bias_data, bias_dims, stride_width,
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
H A Dreference_ops.h160 int stride_width, int stride_height, int pad_width,
183 const int in_x_origin = (out_x * stride_width) - pad_width;
223 const float* bias_data, const Dims<4>& bias_dims, int stride_width,
230 stride_width, stride_height, pad_width, pad_height,
252 int stride_width, int stride_height, int pad_width,
277 const int in_x_origin = (out_x * stride_width) - pad_width;
322 int stride_width, int stride_height, int pad_width,
339 filter_offset, bias_data, bias_dims, stride_width, stride_height,
1686 int stride_width, int stride_height, int pad_width,
1701 const int in_x_origin = (out_x * stride_width)
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
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
[all...]
/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dconv_test.cc47 const TensorData& filter, const TensorData& output, int stride_width = 2,
74 CreateConv2DOptions(builder_, padding, stride_width,
153 /*stride_width=*/3, /*stride_height=*/1);
178 const int stride_width = 1; local
186 {TensorType_FLOAT32, {}}, stride_width, stride_height, padding);
232 const int stride_width = 1; local
240 {TensorType_FLOAT32, {}}, stride_width, stride_height, padding);
286 const int stride_width = 1; local
294 {TensorType_FLOAT32, {}}, stride_width, stride_height, padding,
342 const int stride_width local
[all...]
/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_test.cc420 const int stride_width = 3; local
426 .Attr("strides", {1, stride_height, stride_width, 1})
/external/tensorflow/tensorflow/contrib/lite/
H A Dbuiltin_op_data.h52 int stride_width; member in struct:__anon25865
59 int stride_width; member in struct:__anon25866
71 int stride_width; member in struct:__anon25868
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
H A Dpropagate_fixed_sizes.cc33 int kheight, int stride_width, int stride_height,
44 output_width = (input_width + stride_width - kwidth) / stride_width;
47 output_width = (input_width + stride_width - 1) / stride_width;
55 0, ((output_width - 1) * stride_width + kwidth - input_width) / 2);
168 ComputeConvSizes(input_shape, output_depth, kwidth, kheight, op->stride_width,
224 ComputeConvSizes(input_shape, output_depth, kwidth, kheight, op->stride_width,
700 op->stride_width, op->stride_height, op->padding.type,
717 op->stride_width, o
32 ComputeConvSizes(const Shape& input_shape, int output_depth, int kwidth, int kheight, int stride_width, int stride_height, PaddingType padding_type, Shape* output_shape, FixedPadding* fixed_padding) argument
[all...]
/external/tensorflow/tensorflow/contrib/lite/toco/
H A Dmodel.h357 int stride_width = 0; member in struct:toco::ConvOperator
378 int stride_width = 0; member in struct:toco::DepthwiseConvOperator
638 int stride_width = 0; member in struct:toco::AveragePoolOperator
669 int stride_width = 0; member in struct:toco::MaxPoolOperator
684 int stride_width = 0; member in struct:toco::L2PoolOperator
853 int stride_width = 0; member in struct:toco::TransposeConvOperator

Completed in 629 milliseconds