Home
last modified time | relevance | path

Searched refs:m_inputRows (Results 1 – 5 of 5) sorted by relevance

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorImagePatch.h186 m_inputRows = input_dims[1];
190 m_inputRows = input_dims[NumInputDims-2];
215 m_input_rows_eff = (m_inputRows - 1) * m_row_inflate_strides + 1;
293 m_colInputStride = m_inputDepth * m_inputRows;
294 m_patchInputStride = m_inputDepth * m_inputRows * m_inputCols;
405 if (inputRows[1] < 0 || inputRows[0] >= m_inputRows) {
409 if (inputRows[0] >= 0 && inputRows[1] < m_inputRows) {
489 Index m_inputRows;
DTensorVolumePatch.h200 m_inputRows = input_dims[2];
205 m_inputRows = input_dims[NumInputDims-3];
223 m_input_rows_eff = (m_inputRows - 1) * m_row_inflate_strides + 1;
318 m_colInputStride = m_inputDepth * m_inputRows * m_inputPlanes;
319 m_otherInputStride = m_inputDepth * m_inputRows * m_inputCols * m_inputPlanes;
461 if (inputRows[1] < 0 || inputRows[0] >= m_inputRows) {
562 Index m_inputRows;
/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h66 m_inputRows = input_dims[2]; in CustomTensorEvaluator()
71 m_inputRows = input_dims[NumInputDims - 3]; in CustomTensorEvaluator()
89 m_input_rows_eff = (m_inputRows - 1) * m_row_inflate_strides + 1; in CustomTensorEvaluator()
209 m_colInputStride = m_inputDepth * m_inputRows * m_inputPlanes; in CustomTensorEvaluator()
211 m_inputDepth * m_inputRows * m_inputCols * m_inputPlanes; in CustomTensorEvaluator()
392 if (inputRows[1] < 0 || inputRows[0] >= m_inputRows) { in packet()
592 Index m_inputRows; member
Deigen_spatial_convolutions-inl.h135 m_inputRows = tensor.impl().impl().dimensions()[1]; in TensorContractionInputMapper()
139 m_inputRows = tensor.impl().impl().dimensions()[NumDims - 2]; in TensorContractionInputMapper()
144 m_colInputStride = patch_depth * m_inputRows; in TensorContractionInputMapper()
145 m_patchInputStride = patch_depth * m_inputRows * m_inputCols; in TensorContractionInputMapper()
182 m_inputRows = base_mapper.m_inputRows; in TensorContractionInputMapper()
302 origInputRow >= m_inputRows || in loadCoeff()
328 inputRow >= m_inputRows) { in loadCoeffStandard()
389 if (inputRows[0] >= m_inputRows || inputRows[1] < 0) { in loadPacketStandard()
394 if (inputRows[0] >= 0 && inputRows[1] < m_inputRows) { in loadPacketStandard()
425 inputRow >= m_inputRows) { in loadPacketFast()
[all …]
Deigen_cuboid_convolution.h154 m_inputRows = tensor.impl().impl().dimensions()[2]; in TensorContractionInputMapper()
160 m_inputRows = tensor.impl().impl().dimensions()[NumDims - 3]; in TensorContractionInputMapper()
167 m_colInputStride = m_inputDepth * m_inputRows * m_inputPlanes; in TensorContractionInputMapper()
169 m_inputDepth * m_inputRows * m_inputCols * m_inputPlanes; in TensorContractionInputMapper()
230 m_inputRows = base_mapper.m_inputRows; in TensorContractionInputMapper()
381 origInputCol >= m_inputCols || origInputRow >= m_inputRows || in loadCoeff()
419 inputRow >= m_inputRows || inputPlane < 0 || in loadCoeffStandard()
490 if (inputRows[0] >= m_inputRows || inputRows[1] < 0) { in loadPacketStandard()
551 inputCol >= m_inputCols || inputRow >= m_inputRows || in loadPacketFast()
631 Index m_inputRows; // Number of rows in the input tensor variable
[all …]