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