Home
last modified time | relevance | path

Searched refs:compute_permutation_output_shape (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/core/CPP/kernels/
H A DCPPPermuteKernel.cpp43 …const TensorShape output_shape = misc::shape_calculator::compute_permutation_output_shape(*input, … in validate_arguments()
105 …const TensorShape output_shape = misc::shape_calculator::compute_permutation_output_shape(*input->… in configure()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuPermuteKernel.cpp108 …const TensorShape dst_shape = misc::shape_calculator::compute_permutation_output_shape(*src, perm); in validate_arguments()
246 …const TensorShape dst_shape = misc::shape_calculator::compute_permutation_output_shape(*src, perm); in configure()
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClPermuteKernel.cpp71 …const TensorShape dst_shape = misc::shape_calculator::compute_permutation_output_shape(*src, perm); in validate_arguments()
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClSoftmax.cpp111 …sorShape permuted_shape = misc::shape_calculator::compute_permutation_output_shape(src, … in validate()
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/
H A DCpuSoftmax.cpp144 …sorShape permuted_shape = misc::shape_calculator::compute_permutation_output_shape(*src,… in validate()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h109 inline TensorShape compute_permutation_output_shape(const ITensorInfo &input, const PermutationVect… in compute_permutation_output_shape() function