| /external/tensorflow/tensorflow/core/kernels/ |
| avgpooling_op_gpu.cu.cc | 47 const int kernel_w, const int stride_h,
57 const int phstart = (h < kernel_h) ? 0 : (h - kernel_h) / stride_h + 1;
58 const int phend = min(h / stride_h + 1, pooled_height);
67 int hstart = ph * stride_h - pad_t;
87 const int kernel_w, const int stride_h,
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,
86 const int kernel_h, const int kernel_w, const int stride_h,
|
| pooling_ops_3d_gpu.h | 39 const int kernel_w, const int stride_p, const int stride_h,
|
| maxpooling_op_gpu.cu.cc | 50 // stride_h, stride_w: the strides.
68 const int kernel_w, const int stride_h,
77 int hstart = ph * stride_h - pad_t;
111 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
120 int hstart = ph * stride_h - pad_t;
143 const int kernel_w, const int stride_h,
153 int hstart = (n % pooled_height) * stride_h - pad_t;
183 const int stride_h, const int stride_w, const int pad_t, const int pad_l,
192 int hstart = (n % pooled_height) * stride_h - pad_t;
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,
46 int hstart = ph * stride_h - pad_t;
85 const int stride_p, const int stride_h, const int stride_w, const int pad_p,
97 int hstart = (n % pooled_height) * stride_h - pad_t;
140 const int kernel_w, const int stride_p, const int stride_h,
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 | 47 int stride_h; member in struct:tensorflow::__anon39961::__anon39962
131 int stride_h = param->stride_h; local
146 ij = oj * stride_h - pad_h;
201 int stride_h = stride; local
212 int ofh = (ifh + 2 * pad_h - kh) / stride_h + 1;
263 naive_param.stride_h = stride_h;
|
| avgpooling_op.h | 62 // stride_h: the height of the vertical stride
72 const int kernel_w, const int stride_h,
|
| conv_grad_filter_ops.cc | 62 const int stride_h, const int stride_w, T* col_data) {
63 int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1;
84 h_pad += stride_h;
548 int stride_h = GetTensorDim(strides_, data_format_, 'H'); 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) {
62 int height_col = (height + pad_t + pad_b - filter_h) / stride_h + 1;
85 h_pad += stride_h;
623 int stride_h = GetTensorDim(strides_, data_format_, 'H'); local
629 OP_REQUIRES(context, stride_h > 0 && stride_w > 0,
[all...] |
| conv_ops.cc | 277 const int64 stride_h = GetTensorDim(strides_, data_format_, 'H'); local
283 OP_REQUIRES(context, stride_h > 0 && stride_w > 0,
[all...] |
| /external/tensorflow/tensorflow/python/layers/ |
| convolutional.py | [all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| array_grad.py | 744 _, stride_r, stride_h, _ = op.get_attr("strides")
753 cols_out = int(ceil(cols_in / stride_h))
755 pad_cols = ((cols_out - 1) * stride_h + ksize_c_eff - cols_in) // 2
759 cols_out = int(ceil((cols_in - ksize_c_eff + 1) / stride_h))
761 pad_cols = (cols_out - 1) * stride_h + ksize_c_eff - cols_in
772 col_steps = range(0, cols_out * stride_h, stride_h)
|
| /external/tensorflow/tensorflow/contrib/lite/toco/tflite/ |
| operator.cc | 50 op->stride_height = options.stride_h();
78 op->stride_height = options.stride_h();
106 op->stride_height = options.stride_h();
363 op->stride_height = options.stride_h();
413 op->stride_height = options.stride_h();
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/ |
| model.cc | 268 params->stride_height = conv_params->stride_h();
300 params->stride_height = pool_params->stride_h();
315 params->stride_height = conv_params->stride_h();
|
| /external/tensorflow/tensorflow/contrib/lite/schema/ |
| schema_generated.h | 1339 int32_t stride_h; member in struct:tflite::Conv2DOptionsT
1363 int32_t stride_h() const { function in struct:tflite::FLATBUFFERS_FINAL_CLASS
1429 int32_t stride_h; member in struct:tflite::Pool2DOptionsT
1459 int32_t stride_h() const { function in struct:tflite::FLATBUFFERS_FINAL_CLASS
1543 int32_t stride_h; member in struct:tflite::DepthwiseConv2DOptionsT
1570 int32_t stride_h() const { function in struct:tflite::FLATBUFFERS_FINAL_CLASS
[all...] |
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| layers.py | [all...] |
