Searched refs:src_sizes (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 245 Eigen::DSizes<Eigen::DenseIndex, 5> src_sizes; in launch() local
249 &src_sizes); in launch()
270 Eigen::Tensor<T, 5, Eigen::RowMajor> tensor_out_slice(src_sizes); in launch()
272 tensor_out.tensor<T, 5>().slice(src_indices, src_sizes); in launch()
275 Eigen::Tensor<T, 5, Eigen::RowMajor> out_backprop_slice(src_sizes); in launch()
277 out_backprop.tensor<T, 5>().slice(src_indices, src_sizes); in launch()
420 Eigen::DSizes<Eigen::DenseIndex, 5> src_sizes; in launch() local
424 &src_sizes); in launch()
436 Eigen::Tensor<T, 5, Eigen::RowMajor> slices(src_sizes); in launch()
438 out_backprop.tensor<T, 5>().slice(src_indices, src_sizes); in launch()
|