Lines Matching refs:in_cols
178 int64 in_cols = GetTensorDim(input, data_format, '2');
197 0, (out_cols - 1) * strides[2] + filter_cols - in_cols);
205 const uint64 m = in_batch * in_planes * in_rows * in_cols;
228 filter_cols == in_cols && padding == Padding::VALID &&
233 const uint64 k = in_planes * in_rows * in_cols * in_depth;
267 const int64 new_in_cols = in_cols + cols_odd;
284 in_cols = new_in_cols;
291 FORMAT_NCHW, in_batch, {{in_planes, in_rows, in_cols}}, in_depth);
315 .set_spatial_dim(DimIndex::X, in_cols)
377 {{in_planes, in_rows, in_cols}},