Home
last modified time | relevance | path

Searched refs:Conv2dInfo (Results 1 – 15 of 15) sorted by relevance

/external/ComputeLibrary/src/cpu/operators/
DCpuGemmDirectConv2d.h38 struct Conv2dInfo;
72 …, const ITensorInfo *weights, const ITensorInfo *biases, ITensorInfo *dst, const Conv2dInfo &info);
79 …t ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const Conv2dInfo &info);
DCpuGemmDirectConv2d.cpp78 cpu::AsmGemmInfo init_assembly_metadata(const Conv2dInfo &info, bool is_indirect) in init_assembly_metadata()
110 …c, const ITensorInfo *weights, const ITensorInfo *biases, ITensorInfo *dst, const Conv2dInfo &info) in configure()
156 …st ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const Conv2dInfo &info) in validate()
DCpuConv2d.cpp56 const Conv2dInfo info(conv_info, dilation, act_info, enable_fast_math, num_groups); in configure()
100 const Conv2dInfo info(conv_info, dilation, act_info, enable_fast_math, num_groups); in validate()
134 const Conv2dInfo info(conv_info, dilation, act_info, enable_fast_math, 1); in get_convolution_method()
/external/ComputeLibrary/src/gpu/cl/operators/
DClConv2d.h115 …o *src, ITensorInfo *weights, ITensorInfo *biases, ITensorInfo *dst, const Conv2dInfo &conv2d_info,
123 …sorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const Conv2dInfo &conv2d_info,
140 …TensorInfo *src, const ITensorInfo *weights, const ITensorInfo *dst, const Conv2dInfo &conv2d_info,
DClGemmConv2d.h104 …o *src, ITensorInfo *weights, ITensorInfo *biases, ITensorInfo *dst, const Conv2dInfo &conv2d_info,
112 …Info *weights, const ITensorInfo *biases, const ITensorInfo *output, const Conv2dInfo &conv2d_info,
DClConv2d.cpp81 …o *src, ITensorInfo *weights, ITensorInfo *biases, ITensorInfo *dst, const Conv2dInfo &conv2d_info, in configure()
131 …sorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const Conv2dInfo &conv2d_info, in validate()
179 …TensorInfo *src, const ITensorInfo *weights, const ITensorInfo *dst, const Conv2dInfo &conv2d_info, in get_convolution_method()
DClGemmConv2d.cpp163 const Conv2dInfo &conv2d_info, const WeightsInfo &weights_info) in configure()
349 …sorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const Conv2dInfo &conv2d_info, in validate()
/external/ComputeLibrary/src/runtime/CL/functions/
DCLConvolutionLayer.cpp83 …const Conv2dInfo conv2d_info = Conv2dInfo(conv_info, dilation, act_info, enable_fast_math, num_gro… in configure()
136 …const Conv2dInfo conv2d_info = Conv2dInfo(conv_info, dilation, act_info, enable_fast_math, num_gro… in validate()
166 const Conv2dInfo conv2d_info = Conv2dInfo(conv_info, dilation, act_info, enable_fast_math, 1); in get_convolution_method()
DCLGEMMConvolutionLayer.cpp89 …const Conv2dInfo conv2d_info = Conv2dInfo(conv_info, dilation, act_info, false, num_groups, transf… in configure()
120 …const Conv2dInfo conv2d_info = Conv2dInfo(conv_info, dilation, act_info, false, num_groups, post_o… in validate()
/external/ComputeLibrary/arm_compute/runtime/
DFunctionDescriptors.h56 struct Conv2dInfo struct
58 Conv2dInfo() = default;
60 Conv2dInfo(const PadStrideInfo &conv_info,
/external/ComputeLibrary/arm_compute/runtime/NEON/functions/
DNEGEMMConv2d.h89 …or *input, const ITensor *weights, const ITensor *biases, ITensor *output, const Conv2dInfo &info);
105 …TensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const Conv2dInfo &info);
/external/ComputeLibrary/src/runtime/NEON/functions/
DNEGEMMConv2d.cpp56 …sor *input, const ITensor *weights, const ITensor *biases, ITensor *output, const Conv2dInfo &info) in configure()
72 …ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const Conv2dInfo &info) in validate()
DNEConvolutionLayer.cpp72 const Conv2dInfo info(conv_info, dilation, act_info, enable_fast_math, num_groups); in configure()
110 const Conv2dInfo info(conv_info, dilation, act_info, enable_fast_math, num_groups); in validate()
/external/ComputeLibrary/tests/validation/NEON/
DConvolutionLayer.cpp63 Conv2dInfo conv_info(info, dilation, act_info, false, num_groups); in configure_conv_function()
1124 const auto conv_info = Conv2dInfo{}; in TEST_CASE()
1177 const auto conv_info = Conv2dInfo{}; in TEST_CASE()
/external/ComputeLibrary/utils/
DTypePrinter.h2846 inline ::std::ostream &operator<<(::std::ostream &os, const Conv2dInfo &conv_info)
2863 inline std::string to_string(const Conv2dInfo &conv_info) in to_string()