Searched refs:out_tile_rows (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()65 const int64 row_tiles = (out_rows + out_tile_rows - 1) / out_tile_rows; in GetDeepConvCost()758 const int64 out_tile_rows = transform->output_shape().rows; in operator ()() local760 const int64 out_tile_spatial_size = out_tile_rows * out_tile_cols; in operator ()()807 for (int64 ot_row = 0; ot_row < out_tile_rows; ++ot_row) { in operator ()()950 const int64 out_tile_rows = transform->output_shape().rows; 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 ()()1011 (args.out_rows + out_tile_rows - 1) / out_tile_rows + in operator ()()[all …]