Searched refs:is_batch_dim (Results 1 – 1 of 1) sorted by relevance
197 auto is_batch_dim = [&](int64 d) { in Einsum() local208 if (is_batch_dim(dim_name)) { in Einsum()222 if (!is_batch_dim(dim_name) && !is_contracting(dim_name)) { in Einsum()