Home
last modified time | relevance | path

Searched refs:compute_deep_convolution_shape (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/src/runtime/heuristics/direct_conv/
H A DClDirectConvDefaultConfigBifrost.cpp79 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_G71_f32()
103 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_G71_f16()
127 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_G71_u8()
151 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_default_f32()
175 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_default_f16()
H A DClDirectConvDefaultConfigValhall.cpp81 …orShape dst_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, … in configure_G78_f32()
142 …orShape dst_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, … in configure_G78_f16()
253 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_G78_u8()
278 …orShape dst_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, … in configure_G57_f32()
338 …orShape dst_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, … in configure_G57_f16()
/aosp_15_r20/external/ComputeLibrary/tests/validation/CL/
H A DConvolutionLayer.cpp298 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_SUITE()
325 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_CASE()
346 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_CASE()
367 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_CASE()
388 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_CASE()
411 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_SUITE()
425 …const auto output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_shape, data… in TEST_CASE()
/aosp_15_r20/external/ComputeLibrary/src/runtime/heuristics/indirect_conv/
H A DClIndirectConvDefaultConfigValhall.cpp66 …orShape dst_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, … in configure_G77_f32()
110 …orShape dst_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, … in configure_G77_f16()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuDirectConv2dKernel.cpp81 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *weights, … in validate_arguments()
116 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *weights, … in configure()
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClIndirectConv2dKernel.cpp104 … misc::shape_calculator::compute_deep_convolution_shape(*src, *weights, conv_info)); in validate_arguments()
135 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *weights, … in configure()
H A DClIndirectConv2dAddressPrecalculationKernel.cpp93 …TensorShape output_conv_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *weig… in configure()
H A DClDirectConv2dKernel.cpp127 … misc::shape_calculator::compute_deep_convolution_shape(*src, *weights, conv_info)); in validate_arguments()
176 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *weights, … in configure()
/aosp_15_r20/external/ComputeLibrary/arm_compute/core/utils/misc/
H A DShapeCalculator.h713 inline TensorShape compute_deep_convolution_shape(const TensorShape &input_shape, DataLayout input_… in compute_deep_convolution_shape() function
744 inline TensorShape compute_deep_convolution_shape(const ITensorInfo &input, const ITensorInfo &weig… in compute_deep_convolution_shape() function
746 …return compute_deep_convolution_shape(input.tensor_shape(), input.data_layout(), weights.tensor_sh… in compute_deep_convolution_shape()
769 …TensorShape output_conv2d_shape = compute_deep_convolution_shape(input_shape, input_data_layout, w… in compute_indirect_buffer_shape()
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DIndirectConv2dAddressPrecalculationFixture.h112 …TensorShape output_conv_shape = compute_deep_convolution_shape(shape_conv_src, DataLayout::NHWC, s… in compute_reference()
H A DDirectConvolutionLayerFixture.h70 … const TensorShape output_shape = compute_deep_convolution_shape(input_info, weights_info, info);
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/components/cl/
H A DClComponentDirectConv2d.cpp122 … misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, legacy_pad_stride)); in validate()
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/operators/
H A DGpuConv2d.cpp105 …const auto shape = misc::shape_calculator::compute_deep_convolution_shape(src->tensor_shape(), src… in calculate_and_init_dst_if_empty()
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/operators/
H A DClConv2d.cpp274 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei… in get_convolution_method()
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/gpu/cl/
H A DDirectConv2dFixture.h241 …const TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(input_info… in setup()
/aosp_15_r20/external/ComputeLibrary/tests/validation/NEON/
H A DDirectConvolutionLayer.cpp308 TensorShape output_shape = compute_deep_convolution_shape(input_info, weights_info, info);