Home
last modified time | relevance | path

Searched refs:m_inputPlanes (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h65 m_inputPlanes = input_dims[1]; in CustomTensorEvaluator()
70 m_inputPlanes = input_dims[NumInputDims - 2]; in CustomTensorEvaluator()
88 m_input_planes_eff = (m_inputPlanes - 1) * m_plane_inflate_strides + 1; in CustomTensorEvaluator()
208 m_rowInputStride = m_inputDepth * m_inputPlanes; in CustomTensorEvaluator()
209 m_colInputStride = m_inputDepth * m_inputRows * m_inputPlanes; in CustomTensorEvaluator()
211 m_inputDepth * m_inputRows * m_inputCols * m_inputPlanes; in CustomTensorEvaluator()
412 if (inputPlanes[1] < 0 || inputPlanes[0] >= m_inputPlanes) { in packet()
416 if (inputPlanes[0] >= 0 && inputPlanes[1] < m_inputPlanes) { in packet()
591 Index m_inputPlanes; member
Deigen_cuboid_convolution.h153 m_inputPlanes = tensor.impl().impl().dimensions()[1]; in TensorContractionInputMapper()
159 m_inputPlanes = tensor.impl().impl().dimensions()[NumDims - 2]; in TensorContractionInputMapper()
166 m_rowInputStride = m_inputDepth * m_inputPlanes; in TensorContractionInputMapper()
167 m_colInputStride = m_inputDepth * m_inputRows * m_inputPlanes; in TensorContractionInputMapper()
169 m_inputDepth * m_inputRows * m_inputCols * m_inputPlanes; in TensorContractionInputMapper()
229 m_inputPlanes = base_mapper.m_inputPlanes; in TensorContractionInputMapper()
382 origInputPlane >= m_inputPlanes || in loadCoeff()
420 inputPlane >= m_inputPlanes) { in loadCoeffStandard()
504 if (inputPlanes[0] >= m_inputPlanes || inputPlanes[1] < 0) { in loadPacketStandard()
508 if (inputPlanes[0] >= 0 && inputPlanes[1] < m_inputPlanes) { in loadPacketStandard()
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorVolumePatch.h199 m_inputPlanes = input_dims[1];
204 m_inputPlanes = input_dims[NumInputDims-2];
222 m_input_planes_eff = (m_inputPlanes - 1) * m_plane_inflate_strides + 1;
317 m_rowInputStride = m_inputDepth * m_inputPlanes;
318 m_colInputStride = m_inputDepth * m_inputRows * m_inputPlanes;
319 m_otherInputStride = m_inputDepth * m_inputRows * m_inputCols * m_inputPlanes;
478 if (inputPlanes[1] < 0 || inputPlanes[0] >= m_inputPlanes) {
482 if (inputPlanes[0] >= 0 && inputPlanes[1] < m_inputPlanes) {
561 Index m_inputPlanes;