Lines Matching refs:slice_start
41 inline void setup_slice_coordinates_and_mask(Coordinates &slice_start, int32_t &slice_end_mask, con… in setup_slice_coordinates_and_mask() argument
45 slice_start.set_num_dimensions(input_num_dimensions); in setup_slice_coordinates_and_mask()
49 slice_start.set(k, 0); in setup_slice_coordinates_and_mask()
84 Coordinates slice_start; in configure() local
86 …setup_slice_coordinates_and_mask(slice_start, slice_end_mask, input->info()->tensor_shape().num_di… in configure()
90 slice_start.set(axis_u, slice); in configure()
91 …_vector[slice].configure(compile_context, input, output_vector[slice], slice_start, Coordinates(),… in configure()
104 Coordinates slice_start; in validate() local
108 slice_start.set(wrap_axis(axis, input), k); in validate()
109 …setup_slice_coordinates_and_mask(slice_start, slice_end_mask, input->tensor_shape().num_dimensions… in validate()
110 …ARM_COMPUTE_RETURN_ON_ERROR(CLStridedSlice::validate(input, output_vector[k], slice_start, Coordin… in validate()