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.h501 const Index inputPlanes[2] = {planeIndex + planeOffsets[0], in loadPacketStandard() local
504 if (inputPlanes[0] >= m_inputPlanes || inputPlanes[1] < 0) { in loadPacketStandard()
508 if (inputPlanes[0] >= 0 && inputPlanes[1] < m_inputPlanes) { in loadPacketStandard()
511 depth + inputPlanes[0] * m_planeInputStride + in loadPacketStandard()
1687 const TensorIndex inputPlanes =
1699 out_planes = Eigen::divup(inputPlanes - kernelPlanes + 1,
1708 Eigen::divup(inputPlanes, static_cast<TensorIndex>(stridePlanes));
/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] +