Searched refs:output_shape_size (Results 1 – 1 of 1) sorted by relevance
390 int64 output_shape_size, int64 einsum_threshold_mib, in GetWindowedEinsumConfiguration() argument447 output_shape_size >= einsum_threshold_mib * 1024 * 1024) { in GetWindowedEinsumConfiguration()