Searched refs:stride_rows_ (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | pooling_ops_3d_sycl.h | 48 stride_rows_(stride[1]), in SYCL3DPoolParams() 77 stride_rows_(params.row_stride), in SYCL3DPoolParams() 97 const int stride_rows_; member 148 int rstart = (n % p_.out_rows_) * p_.stride_rows_ - p_.pad_rows_; in operator() 269 (r < p_.window_rows_) ? 0 : (r - p_.window_rows_) / p_.stride_rows_ + 1; 270 const int poolrend = std::min(r / p_.stride_rows_ + 1, p_.out_rows_); 295 int rstart = poolr * p_.stride_rows_ - p_.pad_rows_; 436 int rstart = (n % p_.out_rows_) * p_.stride_rows_ - p_.pad_rows_; 554 int rstart = (n % p_.out_rows_) * p_.stride_rows_ - p_.pad_rows_; 668 (r < p_.window_rows_) ? 0 : (r - p_.window_rows_) / p_.stride_rows_ + 1; [all …]
|
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
D | fused_conv2d_bias_activation_op.cc | 115 stride_rows_ = GetTensorDim(strides, data_format_, 'H'); in FusedConv2DBiasActivationOp() 213 stride_rows_, padding_type_, in Compute() 229 << ", stride_rows = " << stride_rows_ in Compute() 241 filter, stride_rows_, stride_cols_, eigen_padding_type_, in Compute() 247 int32 stride_rows_, stride_cols_; member in tensorflow::FusedConv2DBiasActivationOp
|