Searched refs:m_outputStrides (Results 1 – 10 of 10) sorted by relevance
120 m_outputStrides[0] = 1;122 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];137 m_outputStrides[NumDims-1] = 1;139 m_outputStrides[i] = m_outputStrides[i+1] * m_dimensions[i+1];159 Index patchIndex = index / m_outputStrides[output_stride_index];161 Index patchOffset = index - patchIndex * m_outputStrides[output_stride_index];167 const Index offsetIdx = patchOffset / m_outputStrides[i];168 patchOffset -= offsetIdx * m_outputStrides[i];175 const Index offsetIdx = patchOffset / m_outputStrides[i+1];176 patchOffset -= offsetIdx * m_outputStrides[i+1];[all …]
129 m_outputStrides[0] = 1;132 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];138 m_outputStrides[NumDims-1] = 1;141 m_outputStrides[i] = m_outputStrides[i+1] * m_dimensions[i+1];174 const Index idx0 = indices[0] / m_outputStrides[i];175 const Index idx1 = indices[1] / m_outputStrides[i];178 indices[0] -= idx0 * m_outputStrides[i];179 indices[1] -= idx1 * m_outputStrides[i];185 const Index idx0 = indices[0] / m_outputStrides[i];186 const Index idx1 = indices[1] / m_outputStrides[i];[all …]
119 m_outputStrides[0] = 1;122 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];124 m_outputStrides[NumDims] = m_outputStrides[NumDims-1] * m_dimensions[NumDims-1];127 m_outputStrides[NumDims] = 1;130 m_outputStrides[i+1] = m_outputStrides[i+2] * m_dimensions[i+1];132 m_outputStrides[0] = m_outputStrides[1] * m_dimensions[0];152 const Index idx = index / m_outputStrides[i];157 index -= idx * m_outputStrides[i];165 const Index idx = index / m_outputStrides[i+1];170 index -= idx * m_outputStrides[i+1];[all …]
114 m_outputStrides[0] = 1;117 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];121 m_outputStrides[NumDims-1] = 1;124 m_outputStrides[i] = m_outputStrides[i+1] * m_dimensions[i+1];148 const Index idx = index / m_outputStrides[i];153 index -= idx * m_outputStrides[i];162 const Index idx = index / m_outputStrides[i];167 index -= idx * m_outputStrides[i];220 array<Index, NumDims> m_outputStrides;
131 m_outputStrides[0] = 1;134 m_outputStrides[i] = m_outputStrides[i-1] * m_dimensions[i-1];138 m_outputStrides[NumDims-1] = 1;141 m_outputStrides[i] = m_outputStrides[i+1] * m_dimensions[i+1];175 const Index idx = index / m_outputStrides[i];186 index -= idx * m_outputStrides[i];205 const Index idx = index / m_outputStrides[i];216 index -= idx * m_outputStrides[i];257 const Index idx = index / m_outputStrides[i];268 index -= idx * m_outputStrides[i];[all …]
132 m_outputStrides[0] = 1;135 m_outputStrides[i] = m_outputStrides[i - 1] * m_dimensions[i - 1];139 m_outputStrides[NumDims - 1] = 1;142 m_outputStrides[i] = m_outputStrides[i + 1] * m_dimensions[i + 1];195 const Index idx = index / m_outputStrides[i];197 index -= idx * m_outputStrides[i];202 const Index idx = index / m_outputStrides[i];204 index -= idx * m_outputStrides[i];211 array<Index, NumDims> m_outputStrides;
337 m_outputStrides[0] = 1;339 m_outputStrides[i] = m_outputStrides[i-1] * output_dims[i-1];340 m_fastOutputStrides[i] = internal::TensorIntDivisor<Index>(m_outputStrides[i]);349 m_outputStrides[NumDims-1] = 1;351 m_outputStrides[i] = m_outputStrides[i+1] * output_dims[i+1];352 m_fastOutputStrides[i] = internal::TensorIntDivisor<Index>(m_outputStrides[i]);423 indices[0] -= idx0 * m_outputStrides[i];424 indices[1] -= idx1 * m_outputStrides[i];434 indices[0] -= idx0 * m_outputStrides[i];435 indices[1] -= idx1 * m_outputStrides[i];[all …]
157 m_outputStrides[0] = 1;162 m_outputStrides[j] = m_outputStrides[j-1] * m_dimensions[j-1];167 m_outputStrides[NumDims - 1] = 1;172 m_outputStrides[j] = m_outputStrides[j+1] * m_dimensions[j+1];201 subs[i] = index / m_outputStrides[i];202 index -= subs[i] * m_outputStrides[i];207 subs[i] = index / m_outputStrides[i];208 index -= subs[i] * m_outputStrides[i];282 array<Index, NumDims> m_outputStrides;321 subs[i] = index / this->m_outputStrides[i];[all …]
434 m_outputStrides[0] = 1;436 m_outputStrides[i] = m_outputStrides[i - 1] * m_dimensions[i - 1];439 m_outputStrides.back() = 1;441 m_outputStrides[i] = m_outputStrides[i + 1] * m_dimensions[i + 1];711 const Index idx = index / m_outputStrides[i];713 index -= idx * m_outputStrides[i];724 const Index idx = index / m_outputStrides[i];726 index -= idx * m_outputStrides[i];743 array<Index, NumOutputDims> m_outputStrides;
86 m_outputStrides[i] = outputStrides[ordering[i]]; in IndexMapper()145 outputIndex += idx * m_outputStrides[d]; in mapCudaOutputPlaneToTensorOutputOffset()148 outputIndex += p * m_outputStrides[NumKernelDims]; in mapCudaOutputPlaneToTensorOutputOffset()156 outputIndex += idx * m_outputStrides[d]; in mapCudaOutputPlaneToTensorOutputOffset()159 outputIndex += p * m_outputStrides[limit]; in mapCudaOutputPlaneToTensorOutputOffset()175 return i * m_outputStrides[offset]; in mapCudaOutputKernelToTensorOutputOffset()189 return i * m_outputStrides[offset] + j * m_outputStrides[offset + 1]; in mapCudaOutputKernelToTensorOutputOffset()204 return i * m_outputStrides[offset] + j * m_outputStrides[offset + 1] + in mapCudaOutputKernelToTensorOutputOffset()205 k * m_outputStrides[offset + 2]; in mapCudaOutputKernelToTensorOutputOffset()211 array<Index, NumDims> m_outputStrides; variable