Lines Matching refs:Wrapper

32 struct Wrapper {  struct
33 static Wrapper stride(int64_t v) { in stride() argument
34 return (ShapedType::isDynamicStrideOrOffset(v)) ? Wrapper{true, 0} in stride()
35 : Wrapper{false, v}; in stride()
37 static Wrapper offset(int64_t v) { in offset() argument
38 return (ShapedType::isDynamicStrideOrOffset(v)) ? Wrapper{true, 0} in offset()
39 : Wrapper{false, v}; in offset()
41 static Wrapper size(int64_t v) { in size() argument
42 return (ShapedType::isDynamic(v)) ? Wrapper{true, 0} : Wrapper{false, v}; in size()
51 bool operator==(Wrapper other) { in operator ==() argument
55 bool operator!=(Wrapper other) { return !(*this == other); } in operator !=() argument
56 Wrapper operator+(Wrapper other) { in operator +() argument
58 return Wrapper{true, 0}; in operator +()
59 return Wrapper{false, other.v + v}; in operator +()
61 Wrapper operator*(Wrapper other) { in operator *() argument
63 return Wrapper{true, 0}; in operator *()
64 return Wrapper{false, other.v * v}; in operator *()
1733 using saturated_arith::Wrapper; in computeExpandedLayoutMap()
1735 currentStrideToExpand = (Wrapper::stride(currentStrideToExpand) * in computeExpandedLayoutMap()
1736 Wrapper::size(resultShape[shapeIndex--])) in computeExpandedLayoutMap()
1857 using saturated_arith::Wrapper; in computeCollapsedLayoutMap()
1858 auto stride = Wrapper::stride(resultStrides[resultStrideIndex--]); in computeCollapsedLayoutMap()
1860 stride = stride * Wrapper::size(srcShape[idx]); in computeCollapsedLayoutMap()
1876 auto srcStride = Wrapper::stride(srcStrides[idx - 1]); in computeCollapsedLayoutMap()
1904 using saturated_arith::Wrapper; in computeCollapsedType()
1905 auto groupSize = Wrapper::size(1); in computeCollapsedType()
1907 groupSize = groupSize * Wrapper::size(srcType.getDimSize(srcDim)); in computeCollapsedType()
2108 using saturated_arith::Wrapper; in inferResultType()
2110 (Wrapper::offset(targetOffset) + in inferResultType()
2111 Wrapper::offset(staticOffset) * Wrapper::stride(targetStride)) in inferResultType()
2121 using saturated_arith::Wrapper; in inferResultType()
2123 (Wrapper::stride(sourceStride) * Wrapper::stride(staticStride)) in inferResultType()