Searched refs:inferred_first_dim (Results 1 – 1 of 1) sorted by relevance
342 int64 inferred_first_dim = first_dim; in Compute() local362 if (inferred_first_dim == -1) { in Compute()363 inferred_first_dim = t.shape().dim_size(0); in Compute()364 } else if (inferred_first_dim != t.shape().dim_size(0)) { in Compute()365 inferred_first_dim = -1; in Compute()371 first_dim = inferred_first_dim; in Compute()