Home
last modified time | relevance | path

Searched refs:tensor_info_output (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/
H A DNEFlattenLayer.cpp65 …const TensorInfo tensor_info_output = input->clone()->set_tensor_shape(misc::shape_calculator::com… in validate() local
66 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(output, &tensor_info_output); in validate()
/aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/
H A DCLFlattenLayer.cpp73 …const TensorInfo tensor_info_output = input->clone()->set_tensor_shape(misc::shape_calculator::com… in validate() local
74 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(output, &tensor_info_output); in validate()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEReorgLayerKernel.cpp59 …const TensorInfo tensor_info_output = output->clone()->set_tensor_shape(misc::shape_calculator::co… in validate_arguments() local
60 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(output, &tensor_info_output); in validate_arguments()
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLReorgLayerKernel.cpp59 …const TensorInfo tensor_info_output = output->clone()->set_tensor_shape(misc::shape_calculator::co… in validate_arguments() local
60 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(output, &tensor_info_output); in validate_arguments()
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClWinogradFilterTransformKernel.cpp70 …const TensorInfo tensor_info_output = input->clone()->set_tensor_shape(compute_winograd_filter_tra… in validate_arguments() local
72 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(output, &tensor_info_output); in validate_arguments()
H A DClWinogradOutputTransformKernel.cpp91 …const TensorInfo tensor_info_output = input->clone()->set_tensor_shape(compute_winograd_output_tra… in validate_arguments() local
93 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(output, &tensor_info_output); in validate_arguments()
H A DClIm2ColKernel.cpp86 …const TensorInfo tensor_info_output = dst->clone()->set_tensor_shape(compute_im2col_conv_shape(src… in validate_arguments() local
87 ARM_COMPUTE_RETURN_ERROR_ON_MISMATCHING_SHAPES(dst, &tensor_info_output); in validate_arguments()