Searched refs:stride_i (Results 1 – 3 of 3) sorted by relevance
276 int64& stride_i = (*strides)[i]; in ValidateStridedSliceOp() local278 if (stride_i == 0) { in ValidateStridedSliceOp()290 {stride_i > 0 ? 0 : -1, stride_i > 0 ? dim_i : dim_i - 1}}; in ValidateStridedSliceOp()292 auto canonical = [stride_i, dim_i, masks, valid_range](int64 x, int c) { in ValidateStridedSliceOp()294 return stride_i > 0 ? valid_range[c] : valid_range[(c + 1) & 1]; in ValidateStridedSliceOp()302 if (shrink_i && stride_i <= 0) { in ValidateStridedSliceOp()306 (*is_simple_slice) &= stride_i == 1; in ValidateStridedSliceOp()329 stride_i == 1 && begin_i == 0 && end_i == dim_i; in ValidateStridedSliceOp()331 (*slice_dim0) &= (i == 0 && stride_i == 1) || take_all_in_dimension; in ValidateStridedSliceOp()333 (*is_identity) &= stride_i == 1 && begin_and_end_masked; in ValidateStridedSliceOp()[all …]
1777 int64_t stride_i = stride[i]; in CalculateSlicedShapeFromDenseIndices() local1782 int64_t bounds[] = {stride_i > 0 ? 0 : -1, in CalculateSlicedShapeFromDenseIndices()1783 stride_i > 0 ? dim_i : dim_i - 1}; in CalculateSlicedShapeFromDenseIndices()1788 if (masks[c]) return stride_i > 0 ? bounds[c] : bounds[(c + 1) & 1]; in CalculateSlicedShapeFromDenseIndices()1803 if (interval_len != 0 && (interval_len < 0) == (stride_i < 0)) in CalculateSlicedShapeFromDenseIndices()1804 size_i = (interval_len / stride_i) + (interval_len % stride_i != 0); in CalculateSlicedShapeFromDenseIndices()1815 stride[i] = stride_i; in CalculateSlicedShapeFromDenseIndices()
1738 isl_val *stride_i; in strided_tile() local1742 stride_i = isl_val_copy(bound->stride); in strided_tile()1745 stride_i = isl_val_one(ctx); in strided_tile()1750 stride = isl_multi_val_set_val(stride, i, stride_i); in strided_tile()