Home
last modified time | relevance | path

Searched refs:in_planes_ (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_3d_sycl.h41 in_planes_(in_planes), in SYCL3DPoolParams()
70 in_planes_(params.tensor_in_planes), in SYCL3DPoolParams()
88 const int in_planes_; member
153 int pend = std::min(pstart + p_.window_planes_, p_.in_planes_); in operator()
158 input_data + n * p_.in_planes_ * p_.in_cols_ * p_.in_rows_ * p_.depth_; in operator()
272 const int p = (n % p_.in_planes_) + p_.pad_planes_;
278 n /= p_.in_planes_;
280 index - n * p_.in_planes_ * p_.in_cols_ * p_.in_rows_ * p_.depth_;
283 input_data + n * p_.in_planes_ * p_.in_cols_ * p_.in_rows_ * p_.depth_;
291 const int pend = std::min(pstart + p_.window_planes_, p_.in_planes_);
[all …]