Home
last modified time | relevance | path

Searched refs:stride_w (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Davgpooling_op_gpu.cu.cc49 const int stride_w, const int pad_t, in DEFINE_GPU_KERNELS()
60 const int pwstart = (w < kernel_w) ? 0 : (w - kernel_w) / stride_w + 1; in DEFINE_GPU_KERNELS()
61 const int pwend = min(w / stride_w + 1, pooled_width); in DEFINE_GPU_KERNELS()
69 int wstart = pw * stride_w - pad_l; in DEFINE_GPU_KERNELS()
89 const int stride_w, const int pad_t, in RunAvePoolBackwardNHWC() argument
98 stride_w, pad_t, pad_t, bottom_diff)); 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
78 int wstart = pw * stride_w - pad_l; 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
122 int wstart = pw * stride_w - pad_l; 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
150 int wstart = (n % pooled_width) * stride_w - pad_l; in MaxPoolForwardNHWC()
183 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolBackwardNoMaskNHWC() argument
190 int wstart = (n % pooled_width) * stride_w - pad_l; in MaxPoolBackwardNoMaskNHWC()
270 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolGradBackwardNoMaskNCHW() argument
279 int wstart = pw * stride_w - pad_l; 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
47 int wstart = pw * stride_w - pad_l; in MaxPoolGradBackwardNoMaskNCDHW()
85 const int stride_p, const int stride_h, const int stride_w, const int pad_p, in MaxPoolGradBackwardNoMaskNDHWC() argument
93 int wstart = (n % pooled_width) * stride_w - pad_l; in MaxPoolGradBackwardNoMaskNDHWC()
141 const int stride_w, const int pad_p, const int pad_t, const int pad_l, 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,
88 const int stride_w, const int pad_t, const int pad_l,
Dxsmm_conv2d_test.cc48 int stride_w; member
132 int stride_w = param->stride_w; in naive_conv_fp() local
148 ii = oi * stride_w - pad_w; in naive_conv_fp()
200 int stride_w = stride; in TEST_F() local
213 int ofw = (ifw + 2 * pad_w - kw) / stride_w + 1; in TEST_F()
264 naive_param.stride_w = stride_w; in TEST_F()
Dpooling_ops_3d_gpu.h40 const int stride_w, const int pad_p, const int pad_t,
Davgpooling_op.h73 const int stride_w, const int pad_t,
Dconv_grad_filter_ops.cc68 const int stride_h, const int stride_w, T* col_data) { in Im2col() argument
70 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Im2col()
88 w_pad += stride_w; in Im2col()
456 int stride_w = GetTensorDim(strides_, data_format_, 'W'); 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
70 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Col2im()
90 w_pad += stride_w; in Col2im()
593 int stride_w = GetTensorDim(strides_, data_format_, 'W'); in Conv2DSlowBackpropInputOp() local
598 OP_REQUIRES(context, stride_h > 0 && stride_w > 0, in Conv2DSlowBackpropInputOp()
Ddepthwise_conv_op.cc278 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeOp() local
282 OP_REQUIRES(context, stride_ == stride_w, in DepthwiseConv2dNativeOp()
Ddepthwise_conv_grad_op.cc552 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeBackpropInputOp() local
556 OP_REQUIRES(context, stride_ == stride_w, in DepthwiseConv2dNativeBackpropInputOp()
1026 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeBackpropFilterOp() local
1030 OP_REQUIRES(context, stride_ == stride_w, in DepthwiseConv2dNativeBackpropFilterOp()
Dconv_grad_ops_3d.cc70 const int stride_w, T* im_data) { in Col2im() argument
73 const int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Col2im()
100 w_pad += stride_w; in Col2im()
119 const int stride_w, T* col_data) { in Im2col() argument
122 const int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1; in Im2col()
147 w_pad += stride_w; in Im2col()
/external/tensorflow/tensorflow/lite/kernels/
Dtranspose_conv_test.cc42 const TensorData& output, Padding padding, int stride_w, in TransposeConvOpModel() argument
55 CreateTransposeConvOptions(builder_, padding, stride_w, stride_h) in TransposeConvOpModel()
/external/tensorflow/tensorflow/lite/schema/
Dschema_v0.fbs114 stride_w:int;
121 stride_w:int;
130 stride_w:int;
Dschema_v2.fbs135 stride_w:int;
142 stride_w:int;
151 stride_w:int;
Dschema_v1.fbs127 stride_w:int;
134 stride_w:int;
143 stride_w:int;
Dschema_v3.fbs145 stride_w:int;
152 stride_w:int;
161 stride_w:int;
Dschema.fbs335 stride_w:int;
344 stride_w:int;
354 stride_w:int;
638 stride_w:int;
Dschema_generated.h2771 int32_t stride_w;
2778 stride_w(0),
2799 int32_t stride_w() const {
2835 void add_stride_w(int32_t stride_w) {
2836 fbb_.AddElement<int32_t>(Conv2DOptions::VT_STRIDE_W, stride_w, 0);
2865 int32_t stride_w = 0,
2874 builder_.add_stride_w(stride_w);
2885 int32_t stride_w;
2892 stride_w(0),
2913 int32_t stride_w() const {
[all …]
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.cc157 params->stride_width = conv_params->stride_w(); in ParseOpData()
200 params->stride_width = pool_params->stride_w(); in ParseOpData()
215 params->stride_width = conv_params->stride_w(); in ParseOpData()
574 params->stride_width = transpose_conv_params->stride_w(); in ParseOpData()
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_version.md38 stride_w:int;
56 stride_w:int;
113 params->stride_width = conv_params->stride_w();
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
806 stride=stride_w,
847 stride_h, stride_w = self.strides
867 stride=stride_w,
1064 stride_d, stride_h, stride_w = self.strides
1086 stride=stride_w)
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
1154 stride=stride_w)
/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)
2814 stride_w] if data_format.startswith('NC') else [
2815 1, stride_h, stride_w, 1
/external/tensorflow/tensorflow/lite/toco/tflite/
Doperator.cc58 op->stride_width = options.stride_w(); in ReadOptions()
97 op->stride_width = options.stride_w(); in ReadOptions()
151 op->stride_width = options.stride_w(); in ReadOptions()
655 op->stride_width = options.stride_w(); in ReadOptions()
713 op->stride_width = options.stride_w(); in ReadOptions()
1604 op->stride_width = options.stride_w(); in ReadOptions()

12