Searched refs:stride_cols_ (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | pooling_ops_3d_sycl.h | 49 stride_cols_(stride[0]), in SYCL3DPoolParams() 78 stride_cols_(params.col_stride), in SYCL3DPoolParams() 98 const int stride_cols_; member 144 int cstart = (n % p_.out_cols_) * p_.stride_cols_ - p_.pad_cols_; in operator() 264 (c < p_.window_cols_) ? 0 : (c - p_.window_cols_) / p_.stride_cols_ + 1; 265 const int poolcend = std::min(c / p_.stride_cols_ + 1, p_.out_cols_); 300 int cstart = poolc * p_.stride_cols_ - p_.pad_cols_; 432 int cstart = (n % p_.out_cols_) * p_.stride_cols_ - p_.pad_cols_; 550 int cstart = (n % p_.out_cols_) * p_.stride_cols_ - p_.pad_cols_; 663 (c < p_.window_cols_) ? 0 : (c - p_.window_cols_) / p_.stride_cols_ + 1; [all …]
|
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
D | fused_conv2d_bias_activation_op.cc | 116 stride_cols_ = GetTensorDim(strides, data_format_, 'W'); in FusedConv2DBiasActivationOp() 216 stride_cols_, padding_type_, in Compute() 228 << ", stride_cols = " << stride_cols_ in Compute() 241 filter, stride_rows_, stride_cols_, eigen_padding_type_, in Compute() 247 int32 stride_rows_, stride_cols_; member in tensorflow::FusedConv2DBiasActivationOp
|