Searched refs:isColMajor (Results 1 – 7 of 7) sorted by relevance
109 static const bool isColMajor = variable117 isColMajor ? kern.dimensions()[0] : kern.dimensions()[4];120 isColMajor ? kern.dimensions()[1] : kern.dimensions()[3];122 isColMajor ? kern.dimensions()[2] : kern.dimensions()[2];124 isColMajor ? kern.dimensions()[3] : kern.dimensions()[1];126 isColMajor ? kern.dimensions()[4] : kern.dimensions()[0];129 isColMajor ? out.dimensions()[1] : out.dimensions()[NumDims - 2];131 isColMajor ? out.dimensions()[2] : out.dimensions()[NumDims - 3];133 isColMajor ? out.dimensions()[3] : out.dimensions()[NumDims - 4];174 if (isColMajor) {[all …]
118 static const bool isColMajor = variable126 isColMajor ? kern.dimensions()[0] : kern.dimensions()[3];129 isColMajor ? kern.dimensions()[1] : kern.dimensions()[2];131 isColMajor ? kern.dimensions()[2] : kern.dimensions()[1];133 isColMajor ? kern.dimensions()[3] : kern.dimensions()[0];143 const TensorIndex outputRows = isColMajor146 const TensorIndex outputCols = isColMajor174 typedef typename internal::conditional<isColMajor, ReverseColMajor,179 if (isColMajor) {195 if (isColMajor) {[all …]
82 static const bool isColMajor = (internal::traits<Input>::Layout == ColMajor); variable83 static const int idxRows = isColMajor ? 1 : 2;84 static const int idxCols = isColMajor ? 2 : 1;113 if (isColMajor) {186 static const bool isColMajor = (internal::traits<Input>::Layout == ColMajor); in CuboidMaxPooling() local194 static const int idxPlanes = isColMajor ? 1 : 3; in CuboidMaxPooling()196 static const int idxCols = isColMajor ? 3 : 1; in CuboidMaxPooling()230 if (isColMajor) { in CuboidMaxPooling()439 static const bool isColMajor = (internal::traits<Input>::Layout == ColMajor);440 static const int idxRows = isColMajor ? 1 : 2;[all …]
1663 static const bool isColMajor = (internal::traits<Input>::Layout == ColMajor);1669 isColMajor ? kern.dimensions()[0] : kern.dimensions()[4];1671 isColMajor ? kern.dimensions()[1] : kern.dimensions()[3];1675 isColMajor ? kern.dimensions()[2] : kern.dimensions()[2];1677 isColMajor ? kern.dimensions()[3] : kern.dimensions()[1];1679 isColMajor ? kern.dimensions()[4] : kern.dimensions()[0];1681 if (isColMajor) {1688 isColMajor ? in.dimension(1) : in.dimension(NumDims - 2);1690 isColMajor ? in.dimension(2) : in.dimension(NumDims - 3);1692 isColMajor ? in.dimension(3) : in.dimension(NumDims - 4);[all …]
1377 const bool isColMajor = (internal::traits<Input>::Layout == ColMajor);1384 isColMajor ? kern.dimensions()[0] : kern.dimensions()[3];1387 isColMajor ? kern.dimensions()[1] : kern.dimensions()[2];1389 isColMajor ? kern.dimensions()[2] : kern.dimensions()[1];1391 isColMajor ? kern.dimensions()[3] : kern.dimensions()[0];1402 isColMajor ? in.dimension(1) : in.dimension(NumDims - 2);1404 isColMajor ? in.dimension(2) : in.dimension(NumDims - 3);1431 if (isColMajor) {1452 if (isColMajor) {1469 if (isColMajor) {
116 const bool isColMajor = (internal::traits<Input>::Layout == ColMajor); variable123 isColMajor ? kern.dimensions()[0] : kern.dimensions()[3];126 isColMajor ? kern.dimensions()[1] : kern.dimensions()[2];128 isColMajor ? kern.dimensions()[2] : kern.dimensions()[1];130 isColMajor ? kern.dimensions()[3] : kern.dimensions()[0];141 isColMajor ? in.dimension(1) : in.dimension(NumDims - 2);143 isColMajor ? in.dimension(2) : in.dimension(NumDims - 3);170 if (isColMajor) {191 if (isColMajor) {208 if (isColMajor) {
113 const bool isColMajor = (internal::traits<Input>::Layout == ColMajor); variable120 isColMajor ? kern.dimensions()[0] : kern.dimensions()[3];123 isColMajor ? kern.dimensions()[1] : kern.dimensions()[2];125 isColMajor ? kern.dimensions()[2] : kern.dimensions()[1];127 isColMajor ? kern.dimensions()[3] : kern.dimensions()[0];138 isColMajor ? in.dimension(1) : in.dimension(NumDims - 2);140 isColMajor ? in.dimension(2) : in.dimension(NumDims - 3);167 if (isColMajor) {188 if (isColMajor) {205 if (isColMajor) {