Searched refs:dst_sizes (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 251 Eigen::DSizes<Eigen::DenseIndex, 5> dst_sizes; in launch() local
255 &dst_sizes); in launch()
269 Eigen::Tensor<T, 5, Eigen::RowMajor> tensor_in_slice(dst_sizes); in launch()
271 tensor_in.tensor<T, 5>().slice(dst_indices, dst_sizes); in launch()
285 Eigen::Tensor<T, 5, Eigen::RowMajor> select_slice(dst_sizes); in launch()
286 Eigen::Tensor<T, 5, Eigen::RowMajor> mat0(dst_sizes); in launch()
294 .slice(dst_indices, dst_sizes) in launch()
426 Eigen::DSizes<Eigen::DenseIndex, 5> dst_sizes; in launch() local
429 dst.FillIndicesAndSizes<5>(tensor_in_shape, &dst_indices, &dst_sizes); in launch()
448 .slice(dst_indices, dst_sizes) in launch()
|