Searched refs:max_batch_dim_value (Results 1 – 1 of 1) sorted by relevance
411 int max_batch_dim_value; in FindLeadingShape() local414 max_batch_dim_value = s->dim_size() < 1 ? 1 : s->dim(0).size(); in FindLeadingShape()438 max_batch_dim_value = in FindLeadingShape()439 std::max<int>(max_batch_dim_value, p.shape().dim(0).size()); in FindLeadingShape()447 max_batch_dim_value = result->dim(0).size(); in FindLeadingShape()453 if (max_batch_dim_value <= 1) { in FindLeadingShape()