Searched refs:src_sizes (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | pooling_ops_3d.cc | 249 Eigen::DSizes<Eigen::DenseIndex, 5> src_sizes; in launch() local
253 &src_sizes); in launch()
274 Eigen::Tensor<T, 5, Eigen::RowMajor> tensor_out_slice(src_sizes); in launch()
276 tensor_out.tensor<T, 5>().slice(src_indices, src_sizes); in launch()
279 Eigen::Tensor<T, 5, Eigen::RowMajor> out_backprop_slice(src_sizes); in launch()
281 out_backprop.tensor<T, 5>().slice(src_indices, src_sizes); in launch()
424 Eigen::DSizes<Eigen::DenseIndex, 5> src_sizes; in launch() local
428 &src_sizes); in launch()
440 Eigen::Tensor<T, 5, Eigen::RowMajor> slices(src_sizes); in launch()
442 out_backprop.tensor<T, 5>().slice(src_indices, src_sizes); in launch()
|