Home
last modified time | relevance | path

Searched refs:stride_h (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Davgpooling_op_gpu.cu.cc48 const int kernel_w, const int stride_h, in DEFINE_GPU_KERNELS()
58 const int phstart = (h < kernel_h) ? 0 : (h - kernel_h) / stride_h + 1; in DEFINE_GPU_KERNELS()
59 const int phend = min(h / stride_h + 1, pooled_height); in DEFINE_GPU_KERNELS()
68 int hstart = ph * stride_h - pad_t; in DEFINE_GPU_KERNELS()
88 const int kernel_w, const int stride_h, in RunAvePoolBackwardNHWC() argument
97 channels, pooled_height, pooled_width, kernel_h, kernel_w, stride_h, in RunAvePoolBackwardNHWC()
107 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
113 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
119 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
Dmaxpooling_op_gpu.cu.cc70 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNCHW() argument
77 int hstart = ph * stride_h - pad_t; in MaxPoolForwardNCHW()
112 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C() argument
121 int hstart = ph * stride_h - pad_t; in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
144 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNHWC() argument
152 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolForwardNHWC()
183 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolBackwardNoMaskNHWC() argument
192 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolBackwardNoMaskNHWC()
270 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNCHW() argument
278 int hstart = ph * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNCHW()
[all …]
Dpooling_ops_3d_gpu.cu.cc35 const int stride_p, const int stride_h, const int stride_w, const int pad_p, in MaxPoolGradBackwardNoMaskNCDHW() argument
46 int hstart = ph * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNCDHW()
85 const int stride_p, const int stride_h, const int stride_w, const int pad_p, in MaxPoolGradBackwardNoMaskNDHWC() argument
97 int hstart = (n % pooled_height) * stride_h - pad_t; in MaxPoolGradBackwardNoMaskNDHWC()
140 const int kernel_w, const int stride_p, const int stride_h, in operator ()() argument
151 kernel_w, stride_p, stride_h, stride_w, pad_p, pad_t, pad_l, top_diff, in operator ()()
158 kernel_w, stride_p, stride_h, stride_w, pad_p, pad_t, pad_l, top_diff, in operator ()()
Dmaxpooling_op_gpu.h40 const int kernel_w, const int stride_h, const int stride_w,
50 const int kernel_w, const int stride_h, const int stride_w,
68 const int kernel_w, const int stride_h, const int stride_w,
87 const int kernel_h, const int kernel_w, const int stride_h,
Dxsmm_conv2d_test.cc47 int stride_h; member
131 int stride_h = param->stride_h; in naive_conv_fp() local
146 ij = oj * stride_h - pad_h; in naive_conv_fp()
201 int stride_h = stride; in TEST_F() local
212 int ofh = (ifh + 2 * pad_h - kh) / stride_h + 1; in TEST_F()
263 naive_param.stride_h = stride_h; in TEST_F()
Dpooling_ops_3d_gpu.h39 const int kernel_w, const int stride_p, const int stride_h,
Davgpooling_op.h72 const int kernel_w, const int stride_h,
Dconv_grad_filter_ops.cc68 const int stride_h, const int stride_w, T* col_data) { in Im2col() argument
69 int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1; in Im2col()
90 h_pad += stride_h; in Im2col()
455 int stride_h = GetTensorDim(strides_, data_format_, 'H'); in Conv2DSlowBackpropFilterOp() local
461 OP_REQUIRES(context, stride_h > 0 && stride_w > 0, in Conv2DSlowBackpropFilterOp()
Dconv_grad_input_ops.cc68 const int stride_h, const int stride_w, T* im_data) { in Col2im() argument
69 int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1; in Col2im()
92 h_pad += stride_h; in Col2im()
592 int stride_h = GetTensorDim(strides_, data_format_, 'H'); in Conv2DSlowBackpropInputOp() local
598 OP_REQUIRES(context, stride_h > 0 && stride_w > 0, in Conv2DSlowBackpropInputOp()
Dconv_grad_ops_3d.cc69 const int pad_r, const int stride_p, const int stride_h, in Col2im() argument
72 const int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1; in Col2im()
102 h_pad += stride_h; in Col2im()
118 const int pad_r, const int stride_p, const int stride_h, in Im2col() argument
121 const int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1; in Im2col()
149 h_pad += stride_h; in Im2col()
Dconv_ops.cc306 const int64 stride_h = GetTensorDim(strides, data_format, 'H'); in InitConv2DParameters() local
312 TF_REQUIRES(stride_h > 0 && stride_w > 0, in InitConv2DParameters()
Dmkl_conv_ops.h562 const int64 stride_h = GetTensorDim(strides_, data_format_, 'H'); in MklConvBackpropCommonOp() local
/external/tensorflow/tensorflow/lite/kernels/
Dtranspose_conv_test.cc43 int stride_h) { in TransposeConvOpModel() argument
55 CreateTransposeConvOptions(builder_, padding, stride_w, stride_h) in TransposeConvOpModel()
/external/tensorflow/tensorflow/lite/schema/
Dschema_v0.fbs115 stride_h:int;
122 stride_h:int;
131 stride_h:int;
Dschema_v2.fbs136 stride_h:int;
143 stride_h:int;
152 stride_h:int;
Dschema_v1.fbs128 stride_h:int;
135 stride_h:int;
144 stride_h:int;
Dschema_v3.fbs146 stride_h:int;
153 stride_h:int;
162 stride_h:int;
Dschema.fbs336 stride_h:int;
345 stride_h:int;
355 stride_h:int;
639 stride_h:int;
Dschema_generated.h2772 int32_t stride_h;
2779 stride_h(0),
2802 int32_t stride_h() const {
2838 void add_stride_h(int32_t stride_h) {
2839 fbb_.AddElement<int32_t>(Conv2DOptions::VT_STRIDE_H, stride_h, 0);
2866 int32_t stride_h = 0,
2873 builder_.add_stride_h(stride_h);
2886 int32_t stride_h;
2893 stride_h(0),
2916 int32_t stride_h() const {
[all …]
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc158 params->stride_height = conv_params->stride_h(); in ParseOpData()
201 params->stride_height = pool_params->stride_h(); in ParseOpData()
216 params->stride_height = conv_params->stride_h(); in ParseOpData()
575 params->stride_height = transpose_conv_params->stride_h(); in ParseOpData()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_version.md39 stride_h:int;
57 stride_h:int;
114 params->stride_height = conv_params->stride_h();
Dops_compatibility.md219 stride_w,stride_h: stride of the sliding window
267 stride_w,stride_h: stride of the filter window
284 stride_w,stride_h: stride of the filter window
302 stride_w,stride_h: stride of the filter window
495 stride_w,stride_h: stride of the sliding window
615 stride_w,stride_h: stride of the sliding window
/external/tensorflow/tensorflow/python/keras/layers/
Dconvolutional.py788 stride_h, stride_w = self.strides
800 stride=stride_h,
847 stride_h, stride_w = self.strides
860 stride=stride_h,
1064 stride_d, stride_h, stride_w = self.strides
1081 stride=stride_h)
1090 strides = (1, 1, stride_d, stride_h, stride_w)
1094 strides = (1, stride_d, stride_h, stride_w, 1)
1129 stride_d, stride_h, stride_w = self.strides
1148 stride=stride_h)
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers.py1268 stride_h, stride_w = utils.two_element_tuple(stride)
1283 [1, stride_h, stride_w, 1], padding)
2798 stride_h, stride_w = utils.two_element_tuple(stride)
2813 strides = [1, 1, stride_h,
2815 1, stride_h, stride_w, 1
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc59 op->stride_height = options.stride_h(); in ReadOptions()
98 op->stride_height = options.stride_h(); in ReadOptions()
152 op->stride_height = options.stride_h(); in ReadOptions()
656 op->stride_height = options.stride_h(); in ReadOptions()
714 op->stride_height = options.stride_h(); in ReadOptions()
1605 op->stride_height = options.stride_h(); in ReadOptions()

12