| /external/tensorflow/tensorflow/core/kernels/ |
| avgpooling_op_gpu.cu.cc | 48 const int stride_w, const int pad_t,
59 const int pwstart = (w < kernel_w) ? 0 : (w - kernel_w) / stride_w + 1;
60 const int pwend = min(w / stride_w + 1, pooled_width);
68 int wstart = pw * stride_w - pad_l;
88 const int stride_w, const int pad_t,
96 pooled_height, pooled_width, kernel_h, kernel_w, stride_h, stride_w,
106 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
112 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
|
| maxpooling_op_gpu.h | 40 const int kernel_w, const int stride_h, const int stride_w,
49 const int kernel_w, const int stride_h, const int stride_w,
67 const int kernel_w, const int stride_h, const int stride_w,
87 const int stride_w, const int pad_t, const int pad_l,
|
| pooling_ops_3d_gpu.h | 40 const int stride_w, const int pad_p, const int pad_t,
|
| maxpooling_op_gpu.cu.cc | 50 // stride_h, stride_w: the strides.
69 const int stride_w, const int pad_t,
78 int wstart = pw * stride_w - pad_l;
111 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
121 int wstart = pw * stride_w - pad_l;
144 const int stride_w, const int pad_t,
151 int wstart = (n % pooled_width) * stride_w - pad_l;
183 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
190 int wstart = (n % pooled_width) * stride_w - pad_l;
257 // stride_h, stride_w: the strides
[all...] |
| pooling_ops_3d_gpu.cu.cc | 35 const int stride_p, const int stride_h, const int stride_w, const int pad_p,
47 int wstart = pw * stride_w - pad_l;
85 const int stride_p, const int stride_h, const int stride_w, const int pad_p,
93 int wstart = (n % pooled_width) * stride_w - pad_l;
141 const int stride_w, const int pad_p, const int pad_t, const int pad_l,
151 kernel_w, stride_p, stride_h, stride_w, pad_p, pad_t, pad_l, top_diff,
158 kernel_w, stride_p, stride_h, stride_w, pad_p, pad_t, pad_l, top_diff,
|
| xsmm_conv2d_test.cc | 48 int stride_w; member in struct:tensorflow::__anon39961::__anon39962
132 int stride_w = param->stride_w; local
148 ii = oi * stride_w - pad_w;
200 int stride_w = stride; local
213 int ofw = (ifw + 2 * pad_w - kw) / stride_w + 1;
264 naive_param.stride_w = stride_w;
|
| avgpooling_op.h | 63 // stride_w: the width of the horizontal stride
73 const int stride_w, const int pad_t,
|
| depthwise_conv_op.cc | 273 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); local
277 OP_REQUIRES(context, stride_ == stride_w,
|
| conv_grad_filter_ops.cc | 62 const int stride_h, const int stride_w, T* col_data) {
64 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1;
82 w_pad += stride_w;
549 int stride_w = GetTensorDim(strides_, data_format_, 'W'); local
554 OP_REQUIRES(context, stride_h > 0 && stride_w > 0,
[all...] |
| conv_grad_input_ops.cc | 61 const int stride_h, const int stride_w, T* im_data) {
63 int width_col = (width + pad_l + pad_r - filter_w) / stride_w + 1;
83 w_pad += stride_w;
624 int stride_w = GetTensorDim(strides_, data_format_, 'W'); local
629 OP_REQUIRES(context, stride_h > 0 && stride_w > 0,
[all...] |
| depthwise_conv_grad_op.cc | 538 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); local
542 OP_REQUIRES(context, stride_ == stride_w,
914 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); local
[all...] |
| conv_ops.cc | 278 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); local
283 OP_REQUIRES(context, stride_h > 0 && stride_w > 0,
[all...] |
| /external/tensorflow/tensorflow/python/layers/ |
| convolutional.py | [all...] |
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| operator.cc | 49 op->stride_width = options.stride_w();
77 op->stride_width = options.stride_w();
105 op->stride_width = options.stride_w();
362 op->stride_width = options.stride_w();
412 op->stride_width = options.stride_w();
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/ |
| model.cc | 267 params->stride_width = conv_params->stride_w();
299 params->stride_width = pool_params->stride_w();
314 params->stride_width = conv_params->stride_w();
|
| /external/tensorflow/tensorflow/contrib/lite/schema/ |
| schema_generated.h | 1338 int32_t stride_w; member in struct:tflite::Conv2DOptionsT
1360 int32_t stride_w() const { function in struct:tflite::FLATBUFFERS_FINAL_CLASS
1428 int32_t stride_w; member in struct:tflite::Pool2DOptionsT
1456 int32_t stride_w() const { function in struct:tflite::FLATBUFFERS_FINAL_CLASS
1542 int32_t stride_w; member in struct:tflite::DepthwiseConv2DOptionsT
1567 int32_t stride_w() const { function in struct:tflite::FLATBUFFERS_FINAL_CLASS
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| layers.py | [all...] |
