Searched refs:compute_stack_shape (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/ |
| H A D | CLStackLayerKernel.cpp | 56 …UTE_RETURN_ERROR_ON_MISMATCHING_DIMENSIONS(output->tensor_shape(), compute_stack_shape(*input, axi… in validate_arguments()
67 …auto_init_if_empty(*output, input->clone()->set_tensor_shape(compute_stack_shape(*input, axis, num… in validate_and_configure_window()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/ |
| H A D | NEStackLayerKernel.cpp | 55 …UTE_RETURN_ERROR_ON_MISMATCHING_DIMENSIONS(output->tensor_shape(), compute_stack_shape(*input, axi… in validate_arguments()
66 …auto_init_if_empty(*output, input->clone()->set_tensor_shape(compute_stack_shape(*input, axis, num… in validate_and_configure_window()
|
| /aosp_15_r20/external/ComputeLibrary/src/graph/nodes/ |
| H A D | StackLayerNode.cpp | 60 …output_descriptor.shape = arm_compute::misc::shape_calculator::compute_stack_shape(input_info, axi… in compute_output_descriptor()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | StackLayerFixture.h | 127 …const TensorShape shape_dst = compute_stack_shape(TensorInfo(shape_src, 1, data_type), axis_u, num… in compute_reference()
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/ |
| H A D | ShapeCalculator.h | 1405 inline TensorShape compute_stack_shape(const ITensorInfo &a, unsigned int axis, unsigned int num_te… in compute_stack_shape() function
|