HomeSort by relevance Sort by last modified time
    Searched refs:inputCols (Results 1 - 7 of 7) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
eigen_backward_cuboid_convolutions.h 84 typename internal::traits<OutputBackward>::Index inputCols,
134 Eigen::divup(inputCols, static_cast<TensorIndex>(strideCols));
144 0, (size_x - 1) * strideCols + kernelCols - inputCols);
165 const TensorIndex padding_right = inputCols + kernelCols - 1 -
215 pre_contract_dims[2] = inputRows * inputCols * inputPlanes;
222 pre_contract_dims[0] = inputRows * inputCols * inputPlanes;
250 post_contract_dims[3] = inputCols;
258 post_contract_dims[NumDims - 4] = inputCols;
399 const TensorIndex inputCols =
420 Eigen::divup(inputCols, static_cast<TensorIndex>(strideCols))
    [all...]
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 -
224 pre_contract_dims[1] = inputRows * inputCols;
230 pre_contract_dims[0] = inputRows * inputCols;
253 post_contract_dims[2] = inputCols;
260 post_contract_dims[NumDims - 3] = inputCols;
383 const TensorIndex inputCols =
414 0, (outputCols - 1) * col_stride + kernelColsEff - inputCols);
eigen_cuboid_convolution.h 118 const TensorIndex inputCols =
130 out_width = Eigen::divup(inputCols - kernelCols + 1,
138 out_width = Eigen::divup(inputCols, static_cast<TensorIndex>(strideCols));
eigen_volume_patch.h 369 const Index inputCols[2] = {
372 if (inputCols[1] < 0 || inputCols[0] >= m_inputCols) {
376 if (inputCols[0] != inputCols[1]) {
423 inputCols[0] * m_colInputStride +
eigen_spatial_convolutions.h 327 const Index inputCols[2] = {colIndex + colOffsets[0],
329 if (inputCols[0] >= m_inputCols || inputCols[1] < 0) {
334 if (inputCols[0] == inputCols[1]) {
351 inputCols[0] * m_colInputStride + otherIndex;
    [all...]
  /external/eigen/unsupported/Eigen/CXX11/src/Tensor/
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 + otherIndex * m_patchInputStride;
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 +

Completed in 2054 milliseconds