Searched refs:inputCols (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_cuboid_convolutions.h | 91 typename internal::traits<OutputBackward>::Index inputCols,
149 0, ((outputCols - 1) * col_stride + kernelColsEff - inputCols) / 2);
160 const TensorIndex padding_right = inputCols - (outputCols - 1) * col_stride -
229 pre_contract_dims[1] = inputPlanes * inputRows * inputCols;
236 pre_contract_dims[0] = inputPlanes * inputRows * inputCols;
260 post_contract_dims[3] = inputCols;
268 post_contract_dims[NumDims - 4] = inputCols;
428 const TensorIndex inputCols = variable
462 0, (outputCols - 1) * strideCols + kernelColsEff - inputCols);
474 const auto padded_out_cols = inputCols + kernelCols - 1;
[all …]
|
| D | eigen_backward_spatial_convolutions.h | 101 typename internal::traits<OutputBackward>::Index inputCols,
154 0, ((outputCols - 1) * col_stride + kernelColsEff - inputCols) / 2);
160 const TensorIndex padding_right = inputCols - (outputCols - 1) * col_stride -
229 pre_contract_dims[1] = inputRows * inputCols;
235 pre_contract_dims[0] = inputRows * inputCols;
258 post_contract_dims[2] = inputCols;
265 post_contract_dims[NumDims - 3] = inputCols;
381 const TensorIndex inputCols = variable
412 0, (outputCols - 1) * col_stride + kernelColsEff - inputCols);
|
| D | eigen_volume_patch.h | 370 const Index inputCols[2] = { in packet() local
373 if (inputCols[1] < 0 || inputCols[0] >= m_inputCols) { in packet()
377 if (inputCols[0] != inputCols[1]) { in packet()
424 inputCols[0] * m_colInputStride + in packet()
|
| D | eigen_cuboid_convolution.h | 476 const Index inputCols[2] = {colIndex + colOffsets[0], in loadPacketStandard() local
478 if (inputCols[0] >= m_inputCols || inputCols[1] < 0) { in loadPacketStandard()
482 if (inputCols[0] == inputCols[1]) { in loadPacketStandard()
513 inputCols[0] * m_colInputStride + otherIndex; in loadPacketStandard()
1691 const TensorIndex inputCols =
1703 out_width = Eigen::divup(inputCols - kernelCols + 1,
1711 out_width = Eigen::divup(inputCols, static_cast<TensorIndex>(strideCols));
|
| D | eigen_spatial_convolutions-inl.h | 374 const Index inputCols[2] = {colIndex + colOffsets[0], in loadPacketStandard() local
376 if (inputCols[0] >= m_inputCols || inputCols[1] < 0) { in loadPacketStandard()
381 if (inputCols[0] == inputCols[1]) { in loadPacketStandard()
398 inputCols[0] * m_colInputStride + otherIndex; in loadPacketStandard()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorImagePatch.h | 391 const Index inputCols[2] = {colIndex * m_col_strides + colOffsets[0] -
393 if (inputCols[1] < 0 || inputCols[0] >= m_inputCols) {
397 if (inputCols[0] == inputCols[1]) {
413 …const Index inputIndex = depth + inputRows[0] * m_rowInputStride + inputCols[0] * m_colInputStride…
|
| D | TensorVolumePatch.h | 440 const Index inputCols[2] = {
443 if (inputCols[1] < 0 || inputCols[0] >= m_inputCols) {
447 if (inputCols[0] != inputCols[1]) {
488 inputCols[0] * m_colInputStride +
|