Home
last modified time | relevance | path

Searched refs:inputPlanes (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_cuboid_convolutions.h89 typename internal::traits<OutputBackward>::Index inputPlanes,
145 ((outputPlanes - 1) * plane_stride + kernelPlanesEff - inputPlanes) / 2);
155 const TensorIndex padding_bottom_z = inputPlanes -
229 pre_contract_dims[1] = inputPlanes * inputRows * inputCols;
236 pre_contract_dims[0] = inputPlanes * inputRows * inputCols;
258 post_contract_dims[1] = inputPlanes;
266 post_contract_dims[NumDims - 2] = inputPlanes;
424 const TensorIndex inputPlanes = variable
458 0, (outputPlanes - 1) * stridePlanes + kernelPlanesEff - inputPlanes);
472 const auto padded_out_planes = inputPlanes + kernelPlanes - 1;
[all …]
Deigen_volume_patch.h408 const Index inputPlanes[2] = { in packet() local
412 if (inputPlanes[1] < 0 || inputPlanes[0] >= m_inputPlanes) { in packet()
416 if (inputPlanes[0] >= 0 && inputPlanes[1] < m_inputPlanes) { in packet()
425 m_planeInputStride * inputPlanes[0] + in packet()
Deigen_cuboid_convolution.h473 const Index inputPlanes[2] = {planeIndex + planeOffsets[0], in loadPartialPacketStandard() local
477 inputCol < 0 || inputPlanes[0] >= m_inputPlanes || inputPlanes[1] < 0) { in loadPartialPacketStandard()
480 } else if (inputPlanes[0] >= 0 && inputPlanes[1] < m_inputPlanes) { in loadPartialPacketStandard()
484 const Index inputIndex = depth + inputPlanes[0] * m_planeInputStride + in loadPartialPacketStandard()
567 const Index inputPlanes[2] = {planeIndex + planeOffsets[0], in loadPacketStandardFromSingleColumnSingleRow() local
570 if (inputPlanes[0] >= m_inputPlanes || inputPlanes[1] < 0) { in loadPacketStandardFromSingleColumnSingleRow()
573 if (inputPlanes[0] >= 0 && inputPlanes[1] < m_inputPlanes) { in loadPacketStandardFromSingleColumnSingleRow()
575 const Index inputIndex = depth + inputPlanes[0] * m_planeInputStride + in loadPacketStandardFromSingleColumnSingleRow()
1881 const TensorIndex inputPlanes =
1893 out_planes = Eigen::divup(inputPlanes - kernelPlanes + 1,
[all …]
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorVolumePatch.h474 const Index inputPlanes[2] = {
478 if (inputPlanes[1] < 0 || inputPlanes[0] >= m_inputPlanes) {
482 if (inputPlanes[0] >= 0 && inputPlanes[1] < m_inputPlanes) {
489 m_planeInputStride * inputPlanes[0] +