Compute Library
 21.05
tensor_transform.cpp File Reference

Go to the source code of this file.

Namespaces

 arm_compute
 Copyright (c) 2017-2021 Arm Limited.
 
 arm_compute::helpers
 
 arm_compute::helpers::tensor_transform
 

Functions

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...