Searched refs:src_indices (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 248 Eigen::DSizes<Eigen::DenseIndex, 5> src_indices; in launch() local
252 src.FillIndicesAndSizes<5>(out_backprop.shape(), &src_indices, in launch()
276 tensor_out.tensor<T, 5>().slice(src_indices, src_sizes); in launch()
281 out_backprop.tensor<T, 5>().slice(src_indices, src_sizes); in launch()
423 Eigen::DSizes<Eigen::DenseIndex, 5> src_indices; in launch() local
427 src.FillIndicesAndSizes<5>(out_backprop.shape(), &src_indices, in launch()
442 out_backprop.tensor<T, 5>().slice(src_indices, src_sizes); in launch()
|