Searched refs:m_inputStrides (Results 1 – 8 of 8) sorted by relevance
130 m_inputStrides[0] = 1;133 m_inputStrides[i] = m_inputStrides[i-1] * input_dims[i-1];134 m_inputStrides[i-1] *= op.strides()[i-1];136 m_inputStrides[NumDims-1] *= op.strides()[NumDims-1];139 m_inputStrides[NumDims-1] = 1;142 m_inputStrides[i] = m_inputStrides[i+1] * input_dims[i+1];143 m_inputStrides[i+1] *= op.strides()[i+1];145 m_inputStrides[0] *= op.strides()[0];176 inputIndices[0] += idx0 * m_inputStrides[i];177 inputIndices[1] += idx1 * m_inputStrides[i];[all …]
114 m_inputStrides[0] = 1;117 m_inputStrides[i] = m_inputStrides[i-1] * input_dims[i-1];131 m_inputStrides[NumDims-2] = 1;134 m_inputStrides[i] = m_inputStrides[i+1] * input_dims[i+1];169 inputIndex += (patchIdx + offsetIdx) * m_inputStrides[i];177 inputIndex += (patchIdx + offsetIdx) * m_inputStrides[i];210 inputIndices[0] += (patchIdx[0] + offsetIdx[0]) * m_inputStrides[i];211 inputIndices[1] += (patchIdx[1] + offsetIdx[1]) * m_inputStrides[i];225 inputIndices[0] += (patchIdx[0] + offsetIdx[0]) * m_inputStrides[i];226 inputIndices[1] += (patchIdx[1] + offsetIdx[1]) * m_inputStrides[i];[all …]
130 m_inputStrides[0] = 1;133 m_inputStrides[i] = m_inputStrides[i-1] * input_dims[i-1];137 m_inputStrides[NumDims-1] = 1;140 m_inputStrides[i] = m_inputStrides[i+1] * input_dims[i+1];178 inputIndex += idx * m_inputStrides[i];183 inputIndex += (idx % m_impl.dimensions()[i]) * m_inputStrides[i];208 inputIndex += idx * m_inputStrides[i];213 inputIndex += (idx % m_impl.dimensions()[i]) * m_inputStrides[i];260 inputIndex += idx * m_inputStrides[i];265 inputIndex += (idx % m_impl.dimensions()[i]) * m_inputStrides[i];[all …]
115 m_inputStrides[0] = 1;118 m_inputStrides[i] = m_inputStrides[i-1] * input_dims[i-1];122 m_inputStrides[NumDims-1] = 1;125 m_inputStrides[i] = m_inputStrides[i+1] * input_dims[i+1];152 *inputIndex += idx / m_strides[i] * m_inputStrides[i];166 *inputIndex += idx / m_strides[i] * m_inputStrides[i];221 array<Index, NumDims> m_inputStrides;
331 m_inputStrides[0] = 1;333 m_inputStrides[i] = m_inputStrides[i-1] * input_dims[i-1];343 m_inputStrides[NumDims-1] = 1;345 m_inputStrides[i] = m_inputStrides[i+1] * input_dims[i+1];421 inputIndices[0] += (idx0 + m_offsets[i]) * m_inputStrides[i];422 inputIndices[1] += (idx1 + m_offsets[i]) * m_inputStrides[i];432 inputIndices[0] += (idx0 + m_offsets[i]) * m_inputStrides[i];433 inputIndices[1] += (idx1 + m_offsets[i]) * m_inputStrides[i];468 offset += m_offsets[i] * m_inputStrides[i];473 offset += m_offsets[j] * m_inputStrides[j];[all …]
118 m_inputStrides[0] = 1;121 m_inputStrides[i] = m_inputStrides[i-1] * input_dims[i-1];126 m_inputStrides[NumDims - 1] = 1;129 m_inputStrides[i] = m_inputStrides[i+1] * input_dims[i+1];156 inputIndex += (idx - m_padding[i].first) * m_inputStrides[i];169 inputIndex += (idx - m_padding[i].first) * m_inputStrides[i];283 inputIndex += (idx - m_padding[i].first) * m_inputStrides[i];341 inputIndex += (idx - m_padding[i].first) * m_inputStrides[i];385 array<Index, NumDims> m_inputStrides;
147 m_inputStrides[i] = inputStrides[shuffle[i]];196 inputIndex += idx * m_inputStrides[i];199 return inputIndex + index * m_inputStrides[0];203 inputIndex += idx * m_inputStrides[i];206 return inputIndex + index * m_inputStrides[NumDims - 1];212 array<Index, NumDims> m_inputStrides;
85 m_inputStrides[i] = inputStrides[ordering[i]]; in IndexMapper()121 inputIndex += idx * m_inputStrides[d]; in mapCudaInputPlaneToTensorInputOffset()124 inputIndex += p * m_inputStrides[NumKernelDims]; in mapCudaInputPlaneToTensorInputOffset()132 inputIndex += idx * m_inputStrides[d]; in mapCudaInputPlaneToTensorInputOffset()135 inputIndex += p * m_inputStrides[limit]; in mapCudaInputPlaneToTensorInputOffset()168 return i * m_inputStrides[offset]; in mapCudaInputKernelToTensorInputOffset()182 return i * m_inputStrides[offset] + j * m_inputStrides[offset + 1]; in mapCudaInputKernelToTensorInputOffset()196 return i * m_inputStrides[offset] + j * m_inputStrides[offset + 1] + in mapCudaInputKernelToTensorInputOffset()197 k * m_inputStrides[offset + 2]; in mapCudaInputKernelToTensorInputOffset()210 array<Index, NumDims> m_inputStrides; variable