Searched refs:compute_depthwise_convolution_shape (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
| H A D | CpuDepthwiseConv2dNativeKernel.cpp | 137 …const TensorShape output_shape = misc::shape_calculator::compute_depthwise_convolution_shape(*src,… in validate_arguments()
159 …const TensorShape output_shape = misc::shape_calculator::compute_depthwise_convolution_shape(*src,… in configure()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/ |
| H A D | DepthwiseConvolutionLayerFixture.h | 87 _output_shape = compute_depthwise_convolution_shape(in_info, we_info, info);
402 …const TensorShape dst_shape = compute_depthwise_convolution_shape(TensorInfo(_input_shape, 1, … in compute_reference()
589 …const TensorShape dst_shape = compute_depthwise_convolution_shape(TensorInfo(_input_shape, 1, … in compute_reference()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/internal/ |
| H A D | CpuDepthwiseConv2dAssemblyWrapperKernel.cpp | 217 const TensorShape dst_shape = compute_depthwise_convolution_shape(*src, *weights, info); in configure()
297 …const TensorShape dst_shape = misc::shape_calculator::compute_depthwise_convolution_shape(*src, *w… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/ |
| H A D | CLDepthwiseConvolutionLayerNativeKernel.cpp | 85 …pe output_shape = arm_compute::misc::shape_calculator::compute_depthwise_convolution_shape(*in… in validate_arguments()
194 …rShape output_shape = arm_compute::misc::shape_calculator::compute_depthwise_convolution_shape(*(i… in configure()
|
| /aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/components/cl/ |
| H A D | ClComponentDepthwiseConv2d.cpp | 160 …nst TensorShape output_shape = misc::shape_calculator::compute_depthwise_convolution_shape(*sr… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/ |
| H A D | DepthwiseConv2dFixture.h | 84 …nst TensorShape output_shape = misc::shape_calculator::compute_depthwise_convolution_shape(in_… in setup()
|
| /aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/operators/ |
| H A D | GpuDepthwiseConv2d.cpp | 58 …const TensorShape shape = misc::shape_calculator::compute_depthwise_convolution_shape(*src, *w… in calculate_and_init_dst_if_empty()
|
| /aosp_15_r20/external/ComputeLibrary/src/runtime/CL/functions/ |
| H A D | CLDepthwiseConvolutionLayer.cpp | 207 …rShape permuted_output_shape = shape_calculator::compute_depthwise_convolution_shape(*in… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/src/cpu/operators/ |
| H A D | CpuDepthwiseConv2d.cpp | 342 …nsorShape permuted_output_shape = misc::shape_calculator::compute_depthwise_convolution_shape(*sr… in validate()
|
| /aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/ |
| H A D | ShapeCalculator.h | 421 inline TensorShape compute_depthwise_convolution_shape(const ITensorInfo &input, const ITensorInfo … in compute_depthwise_convolution_shape() function
|