Searched refs:inputCols (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions.h | 102 typename internal::traits<OutputBackward>::Index inputCols,
155 0, ((outputCols - 1) * col_stride + kernelColsEff - inputCols) / 2);
161 const TensorIndex padding_right = inputCols - (outputCols - 1) * col_stride -
230 pre_contract_dims[1] = inputRows * inputCols;
236 pre_contract_dims[0] = inputRows * inputCols;
259 post_contract_dims[2] = inputCols;
266 post_contract_dims[NumDims - 3] = inputCols;
433 const TensorIndex inputCols = variable
470 0, (outputCols - 1) * col_stride + kernelColsEff - inputCols);
480 const TensorIndex padded_out_cols = inputCols + kernelColsEff - 1;
[all …]
|
| 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_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_spatial_convolutions-inl.h | 453 const Index inputCols[]) const { in loadPacketStandardFromSingleColumn() argument
470 inputCols[0] * m_colInputStride + otherIndex; in loadPacketStandardFromSingleColumn()
503 const Index inputCols[2] = {colIndex + colOffsets[0], in loadPacketStandard() local
506 if (inputCols[0] >= m_inputCols || inputCols[1] < 0) { in loadPacketStandard()
510 if (inputCols[0] == inputCols[1]) { in loadPacketStandard()
513 colOffsets, inputCols); in loadPacketStandard()
551 const Index inputCols[2] = {colIndex + colOffsets[0], in loadPacketStandard() local
554 if (inputCols[0] >= m_inputCols || inputCols[1] < 0) { in loadPacketStandard()
558 if (inputCols[0] == inputCols[1]) { in loadPacketStandard()
561 colOffsets, inputCols); in loadPacketStandard()
[all …]
|
| D | eigen_cuboid_convolution.h | 557 const Index rowOffsets[], const Index inputCols[], in loadPacketStandardFromSingleColumnSingleRow() argument
577 inputCols[0] * m_colInputStride + otherIndex; in loadPacketStandardFromSingleColumnSingleRow()
615 const Index inputCols[2] = {colIndex + colOffsets[0], in loadPacketStandard() local
617 if (inputCols[0] >= m_inputCols || inputCols[1] < 0) { in loadPacketStandard()
621 if (inputCols[0] == inputCols[1]) { in loadPacketStandard()
636 colOffsets, rowOffsets, inputCols, inputRows); in loadPacketStandard()
682 const Index inputCols[2] = {colIndex + colOffsets[0], in loadPacketStandard() local
684 if (inputCols[0] >= m_inputCols || inputCols[1] < 0) { in loadPacketStandard()
688 if (inputCols[0] == inputCols[1]) { in loadPacketStandard()
703 colOffsets, rowOffsets, inputCols, inputRows); in loadPacketStandard()
[all …]
|
| /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 +
|