Searched refs:isColMajor (Results 1 – 8 of 8) 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 …]
119 static const bool isColMajor = variable127 isColMajor ? kern.dimensions()[0] : kern.dimensions()[3];130 isColMajor ? kern.dimensions()[1] : kern.dimensions()[2];132 isColMajor ? kern.dimensions()[2] : kern.dimensions()[1];134 isColMajor ? kern.dimensions()[3] : kern.dimensions()[0];144 const TensorIndex outputRows = isColMajor147 const TensorIndex outputCols = isColMajor175 typedef typename internal::conditional<isColMajor, ReverseColMajor,180 if (isColMajor) {196 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()442 static const bool isColMajor = (internal::traits<Input>::Layout == ColMajor);443 static const int idxRows = isColMajor ? 1 : 2;[all …]
1857 static const bool isColMajor = (internal::traits<Input>::Layout == ColMajor);1863 isColMajor ? kern.dimensions()[0] : kern.dimensions()[4];1865 isColMajor ? kern.dimensions()[1] : kern.dimensions()[3];1869 isColMajor ? kern.dimensions()[2] : kern.dimensions()[2];1871 isColMajor ? kern.dimensions()[3] : kern.dimensions()[1];1873 isColMajor ? kern.dimensions()[4] : kern.dimensions()[0];1875 if (isColMajor) {1882 isColMajor ? in.dimension(1) : in.dimension(NumDims - 2);1884 isColMajor ? in.dimension(2) : in.dimension(NumDims - 3);1886 isColMajor ? in.dimension(3) : in.dimension(NumDims - 4);[all …]
1620 const bool isColMajor = (internal::traits<Input>::Layout == ColMajor);1627 isColMajor ? kern.dimensions()[0] : kern.dimensions()[3];1630 isColMajor ? kern.dimensions()[1] : kern.dimensions()[2];1632 isColMajor ? kern.dimensions()[2] : kern.dimensions()[1];1634 isColMajor ? kern.dimensions()[3] : kern.dimensions()[0];1645 isColMajor ? in.dimension(1) : in.dimension(NumDims - 2);1647 isColMajor ? in.dimension(2) : in.dimension(NumDims - 3);1680 if (isColMajor) {1701 if (isColMajor) {1718 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) {
8575 bool isColMajor = matrixIsColMajor[m] != isTransposed; in FillBuffer() local8577 uint32_t arraySize = isColMajor ? cols : rows; in FillBuffer()8578 uint32_t arrayElementComponents = isColMajor ? rows : cols; in FillBuffer()8590 uint32_t row = isColMajor ? c : i; in FillBuffer()8591 uint32_t col = isColMajor ? i : c; in FillBuffer()
16245 bool isColMajor = isColMajorArg->getSExtValue(); in EmitNVPTXBuiltinExpr() local16247 unsigned IID = isColMajor ? II.IID_col : II.IID_row; in EmitNVPTXBuiltinExpr()16290 bool isColMajor = isColMajorArg->getSExtValue(); in EmitNVPTXBuiltinExpr() local16292 unsigned IID = isColMajor ? II.IID_col : II.IID_row; in EmitNVPTXBuiltinExpr()