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