| 576 | } |
| 577 | |
| 578 | VARP _StridedSlice(VARP input, VARP begin, VARP end, VARP strided, int32_t beginMask, |
| 579 | int32_t endMask, int32_t ellipsisMask, int32_t newAxisMask, int32_t shrinkAxisMask) { |
| 580 | std::unique_ptr<OpT> op(new OpT); |
| 581 | op->type = OpType_StridedSlice; |
| 582 | op->main.type = OpParameter_StridedSliceParam; |
| 583 | op->main.value = new StridedSliceParamT; |
| 584 | |
| 585 | op->main.AsStridedSliceParam()->T = DataType_DT_FLOAT; |
| 586 | op->main.AsStridedSliceParam()->beginMask = beginMask; |
| 587 | op->main.AsStridedSliceParam()->endMask = endMask; |
| 588 | op->main.AsStridedSliceParam()->ellipsisMask = ellipsisMask; |
| 589 | op->main.AsStridedSliceParam()->newAxisMask = newAxisMask; |
| 590 | op->main.AsStridedSliceParam()->shrinkAxisMask = shrinkAxisMask; |
| 591 | return (Variable::create(Expr::create(op.get(), {input, begin, end, strided}))); |
| 592 | } |
| 593 | |
| 594 | VARP _StridedSliceWrite(VARP input, VARP begin, VARP end, VARP strided, VARP write, int32_t beginMask, |
| 595 | int32_t endMask, int32_t ellipsisMask, int32_t newAxisMask, int32_t shrinkAxisMask) { |