Searched refs:compute_split_shape (Results 1 – 2 of 2) sorted by relevance
82 …output_shape = arm_compute::misc::shape_calculator::compute_split_shape(input, axis, outputs.size(… in validate()164 … arm_compute::misc::shape_calculator::compute_split_shape(input->info(), axis, _num_outputs)); in configure()
1130 inline TensorShape compute_split_shape(const ITensorInfo *input, unsigned int axis, unsigned int nu… in compute_split_shape() function