Home
last modified time | relevance | path

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

/external/ComputeLibrary/src/runtime/NEON/functions/
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()
/external/ComputeLibrary/src/runtime/CL/functions/
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()
/external/ComputeLibrary/src/core/NEON/kernels/
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()
/external/ComputeLibrary/src/core/CL/kernels/
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()
/external/ComputeLibrary/src/gpu/cl/kernels/
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()
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()
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()