Searched refs:output_backward_shuffle (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions.h | 491 array<TensorIndex, 4> output_backward_shuffle; variable
495 output_backward_shuffle = {3, 1, 2, 0};
499 output_backward_shuffle = {3, 1, 2, 0};
579 output_backward.shuffle(output_backward_shuffle).eval();
|
| D | eigen_backward_cuboid_convolutions.h | 486 array<TensorIndex, 5> output_backward_shuffle; variable
490 output_backward_shuffle = {4, 1, 2, 3, 0};
494 output_backward_shuffle = {4, 1, 2, 3, 0};
578 output_backward.shuffle(output_backward_shuffle).eval();
|