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,
|