Home
last modified time | relevance | path

Searched refs:compute_split_shape (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/arm_compute/runtime/CPP/functions/
H A DCPPSplit.h82 …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()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h1130 inline TensorShape compute_split_shape(const ITensorInfo *input, unsigned int axis, unsigned int nu… in compute_split_shape() function