Searched refs:cur_dim (Results 1 – 2 of 2) sorted by relevance
85 for (int64 cur_dim : LayoutUtil::MinorToMajor(shape)) { in PartitionShapeByMiddleDimensions() local88 bool cur_dim_in_middle = absl::c_linear_search(dims_middle, cur_dim); in PartitionShapeByMiddleDimensions()99 values[cur_segment] *= shape.dimensions(cur_dim); in PartitionShapeByMiddleDimensions()
536 int64_t cur_dim = dim_pair.value().getSExtValue();537 if (cur_dim < -rank || cur_dim >= rank)