Searched refs:out_tile_cols (Results 1 – 1 of 1) sorted by relevance
49 int out_tile_rows, int out_tile_cols, int in_depth, in GetDeepConvCost() argument60 const int64 output_tile_spatial_size = out_tile_rows * out_tile_cols; in GetDeepConvCost()66 const int64 col_tiles = (out_cols + out_tile_cols - 1) / out_tile_cols; in GetDeepConvCost()759 const int64 out_tile_cols = transform->output_shape().cols; in operator ()() local760 const int64 out_tile_spatial_size = out_tile_rows * out_tile_cols; in operator ()()811 for (int64 ot_col = 0; ot_col < out_tile_cols; ++ot_col) { in operator ()()816 const int64 out_buf_index = ot_row * out_tile_cols + ot_col; in operator ()()951 const int64 out_tile_cols = transform->output_shape().cols; in operator ()() local952 const int64 out_tile_spatial_size = out_tile_rows * out_tile_cols; in operator ()()1006 out_depth, out_tile_rows, out_tile_cols, filter_shards_row, in operator ()()[all …]