|
int | calculate_stride_on_index (int index, Coordinates strides) |
| Computes stride of a given index. More...
|
|
int | calculate_start_on_index (TensorShape input_shape, int index, Coordinates starts, Coordinates strides, int32_t begin_mask) |
| Computes absolute start position of a given index for a strided slice operation. More...
|
|
int | calculate_end_on_index (TensorShape input_shape, int index, int start_on_index, Coordinates ends, Coordinates strides, int32_t end_mask=0, int32_t shrink_axis_mask=0) |
| Returns the absolute end position of a given index for a strided slice operation. More...
|
|
std::tuple< Coordinates, Coordinates, Coordinates > | calculate_strided_slice_coords (TensorShape input_shape, Coordinates starts, Coordinates ends, Coordinates strides, int32_t begin_mask=0, int32_t end_mask=0, int32_t shrink_axis_mask=0) |
| Calculate start, end and stride coordinates for a strided slice. More...
|
|
TensorShape | compute_strided_slice_output_shape (TensorShape input_shape, Coordinates starts, Coordinates ends, Coordinates strides, int32_t begin_mask=0, int32_t end_mask=0, int32_t shrink_axis_mask=0, bool return_unshrinked=false) |
| Computes output shape of strided slice. More...
|
|
int32_t | construct_slice_end_mask (Coordinates ends) |
| Constructs end mask in case we want to perform a slice operation using the strided slice interface. More...
|
|