Home
last modified time | relevance | path

Searched refs:isColMajor (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_cuboid_convolutions.h109 static const bool isColMajor = variable
117 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 …]
Deigen_backward_spatial_convolutions.h118 static const bool isColMajor = variable
126 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 = isColMajor
146 const TensorIndex outputCols = isColMajor
174 typedef typename internal::conditional<isColMajor, ReverseColMajor,
179 if (isColMajor) {
195 if (isColMajor) {
[all …]
Deigen_pooling.h82 static const bool isColMajor = (internal::traits<Input>::Layout == ColMajor); variable
83 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() local
194 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 …]
Deigen_cuboid_convolution.h1663 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 …]
Deigen_spatial_convolutions-inl.h1377 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) {
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h116 const bool isColMajor = (internal::traits<Input>::Layout == ColMajor); variable
123 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) {
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h113 const bool isColMajor = (internal::traits<Input>::Layout == ColMajor); variable
120 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) {