Home
last modified time | relevance | path

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

/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorVolumePatch.h69 m_plane_strides(plane_strides), m_row_strides(row_strides), m_col_strides(col_strides),
84 m_plane_strides(plane_strides), m_row_strides(row_strides), m_col_strides(col_strides),
98 DenseIndex plane_strides() const { return m_plane_strides; }
143 const DenseIndex m_plane_strides;
209 m_plane_strides = op.plane_strides();
230 …top_z() + op.padding_bottom_z() - m_patch_planes_eff + 1.f) / static_cast<float>(m_plane_strides));
240 …umext::ceil((m_input_planes_eff - m_patch_planes_eff + 1.f) / static_cast<float>(m_plane_strides));
248 m_outputPlanes = numext::ceil(m_input_planes_eff / static_cast<float>(m_plane_strides));
251 … const Index dz = m_outputPlanes * m_plane_strides + m_patch_planes_eff - 1 - m_input_planes_eff;
389 …const Index inputPlane = planeIndex * m_plane_strides + planeOffset * m_in_plane_strides - m_plane…
[all …]
/external/tensorflow/tensorflow/core/kernels/
Deigen_volume_patch.h75 m_plane_strides = op.plane_strides(); in CustomTensorEvaluator()
103 m_plane_strides); in CustomTensorEvaluator()
122 m_input_planes_eff - m_patch_planes_eff + 1, m_plane_strides); in CustomTensorEvaluator()
132 m_outputPlanes = Eigen::divup(m_input_planes_eff, m_plane_strides); in CustomTensorEvaluator()
136 0, (m_outputPlanes - 1) * m_plane_strides + m_patch_planes_eff - in CustomTensorEvaluator()
305 const Index inputPlane = planeIndex * m_plane_strides + in coeff()
409 planeIndex * m_plane_strides + planeOffsets[0] - m_planePaddingTop, in packet()
410 planeIndex * m_plane_strides + planeOffsets[1] - m_planePaddingTop}; in packet()
451 Index userPlaneStride() const { return m_plane_strides; } in userPlaneStride()
509 const Index inputPlane = planeIndex * m_plane_strides + in coeff()
[all …]
Deigen_cuboid_convolution.h139 m_plane_strides = tensor.impl().userPlaneStride(); in TensorContractionInputMapper()
237 m_plane_strides = base_mapper.m_plane_strides; in TensorContractionInputMapper()
603 planeIndex = planeIndex * m_plane_strides - m_planePaddingTop; in computeBaseIndices()
639 Index m_plane_strides; // User specified plane stride variable