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