Searched refs:max_batch_dim_value (Results 1 – 1 of 1) sorted by relevance
406 int max_batch_dim_value; in FindLeadingShape() local409 max_batch_dim_value = s->dim_size() < 1 ? 1 : s->dim(0).size(); in FindLeadingShape()433 max_batch_dim_value = in FindLeadingShape()434 std::max<int>(max_batch_dim_value, p.shape().dim(0).size()); in FindLeadingShape()442 max_batch_dim_value = result->dim(0).size(); in FindLeadingShape()448 if (max_batch_dim_value <= 1) { in FindLeadingShape()