Searched refs:stridePlanes (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | eigen_pooling.h | 182 DenseIndex stridePlanes, DenseIndex strideRows, in CuboidMaxPooling() argument 211 stridePlanes); in CuboidMaxPooling() 220 static_cast<DenseIndex>(in.dimension(idxPlanes)), stridePlanes); in CuboidMaxPooling() 250 .extract_volume_patches(patchPlanes, patchRows, patchCols, stridePlanes, in CuboidMaxPooling() 538 DenseIndex stridePlanes, DenseIndex strideRows, 566 stridePlanes); 575 static_cast<DenseIndex>(in.dimension(idxPlanes)), stridePlanes); 609 .extract_volume_patches(patchPlanes, patchRows, patchCols, stridePlanes,
|
D | eigen_backward_cuboid_convolutions.h | 394 const DenseIndex stridePlanes = 1, const DenseIndex strideRows = 1, 458 0, (outputPlanes - 1) * stridePlanes + kernelPlanesEff - inputPlanes); 469 const auto expanded_out_planes = (outputPlanes - 1) * stridePlanes + 1; 590 stridePlanes, strideRows, strideCols, 598 stridePlanes, strideRows, strideCols, top_pad_planes,
|
D | eigen_cuboid_convolution.h | 1647 const Index stridePlanes = 1, const Index strideRows = 1, 1700 static_cast<TensorIndex>(stridePlanes)); 1708 Eigen::divup(inputPlanes, static_cast<TensorIndex>(stridePlanes)); 1784 kernelPlanes, kernelRows, kernelCols, stridePlanes, 1791 stridePlanes, strideRows, strideCols,
|