Searched refs:out_tile_spatial_size (Results 1 – 1 of 1) sorted by relevance
760 const int64 out_tile_spatial_size = out_tile_rows * out_tile_cols; in operator ()() local763 ConstMatrixMap A(out_transform_matrix, out_tile_spatial_size, in operator ()()766 MatrixMap C(out_transform_buffer, out_tile_spatial_size, out_buf_stride); in operator ()()952 const int64 out_tile_spatial_size = out_tile_rows * out_tile_cols; in operator ()() local997 TensorShape({out_tile_spatial_size, in operator ()()1003 out_tile_spatial_size, tile_spatial_size, output_transform_matrix); in operator ()()1019 const int64 out_tile_spatial_size = out_tile_rows * out_tile_cols; in operator ()() local1029 out_tile_spatial_size * tile_spatial_size; in operator ()()1044 out_tile_spatial_size * out_depth * filter_shard_size); in operator ()()1077 out_tile_spatial_size * num_tiles * out_depth * filter_shard_size; in operator ()()