Searched refs:dims_middle (Results 1 – 1 of 1) sorted by relevance
79 const Shape& shape, absl::Span<const int64> dims_middle) { in PartitionShapeByMiddleDimensions() argument80 CHECK(LayoutUtil::AreDimensionsConsecutive(shape.layout(), dims_middle)); in PartitionShapeByMiddleDimensions()88 bool cur_dim_in_middle = absl::c_linear_search(dims_middle, cur_dim); in PartitionShapeByMiddleDimensions()