| 336 | } |
| 337 | |
| 338 | af_err af_set_array_indexer(af_index_t* indexer, const af_array idx, |
| 339 | const dim_t dim) { |
| 340 | try { |
| 341 | ARG_ASSERT(0, (indexer != NULL)); |
| 342 | ARG_ASSERT(1, (idx != NULL)); |
| 343 | ARG_ASSERT(2, (dim >= 0 && dim <= 3)); |
| 344 | indexer[dim] = af_index_t{{idx}, false, false}; |
| 345 | } |
| 346 | CATCHALL; |
| 347 | return AF_SUCCESS; |
| 348 | } |
| 349 | |
| 350 | af_err af_set_seq_indexer(af_index_t* indexer, const af_seq* idx, |
| 351 | const dim_t dim, const bool is_batch) { |