HomeSort by relevance Sort by last modified time
    Searched refs:kernel_shuffle (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
eigen_backward_cuboid_convolutions.h 547 array<TensorIndex, 5> kernel_shuffle; local
549 kernel_shuffle[0] = 1;
550 kernel_shuffle[1] = 0;
551 kernel_shuffle[2] = 2;
552 kernel_shuffle[3] = 3;
553 kernel_shuffle[4] = 4;
555 kernel_shuffle[0] = 0;
556 kernel_shuffle[1] = 1;
557 kernel_shuffle[2] = 2;
558 kernel_shuffle[3] = 4
    [all...]
eigen_backward_spatial_convolutions.h 193 array<TensorIndex, 4> kernel_shuffle; local
195 kernel_shuffle[0] = 0;
196 kernel_shuffle[1] = 2;
197 kernel_shuffle[2] = 3;
198 kernel_shuffle[3] = 1;
200 kernel_shuffle[0] = 2;
201 kernel_shuffle[1] = 0;
202 kernel_shuffle[2] = 1;
203 kernel_shuffle[3] = 3;
269 .shuffle(kernel_shuffle)
    [all...]

Completed in 199 milliseconds