Searched refs:cur_dim (Results 1 – 2 of 2) sorted by relevance
42 int64_t cur_dim = dim_pair.value().getSExtValue(); in VerifyReductionInputAndDims() local43 if (cur_dim < -rank || cur_dim >= rank) in VerifyReductionInputAndDims()
58 for (int64_t cur_dim : LayoutUtil::MinorToMajor(shape)) { in PartitionShapeByMiddleDimensions() local61 bool cur_dim_in_middle = absl::c_linear_search(dims_middle, cur_dim); in PartitionShapeByMiddleDimensions()72 values[cur_segment] *= shape.dimensions(cur_dim); in PartitionShapeByMiddleDimensions()