Given the indices of the src-layout, infer the dst index. Parameters ---------- index: Array of Expr The indices in src-layout. Returns ------- dst_index: Array of Expr The inferred indices in dst-layout.
(self, index)
| 104 | """ |
| 105 | |
| 106 | def forward_index(self, index): |
| 107 | """Given the indices of the src-layout, infer the dst index. |
| 108 | |
| 109 | Parameters |
| 110 | ---------- |
| 111 | index: Array of Expr |
| 112 | The indices in src-layout. |
| 113 | |
| 114 | Returns |
| 115 | ------- |
| 116 | dst_index: Array of Expr |
| 117 | The inferred indices in dst-layout. |
| 118 | """ |
| 119 | return _ffi_api.SBijectiveLayoutForwardIndex(self, index) # type: ignore |
| 120 | |
| 121 | def backward_index(self, index): |
| 122 | """Given the indices of the dst-layout, infer the src index. |
no outgoing calls