Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_pooling.h182 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,
Deigen_backward_cuboid_convolutions.h394 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,
Deigen_cuboid_convolution.h1647 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,