Searched refs:conv_method (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/ComputeLibrary/src/graph/mutators/ |
| H A D | NodeFusionMutator.cpp | 107 const auto conv_method = conv_node->convolution_method(); in fuse_convolution_with_batch_normalization() local
119 …usedConvolutionBatchNormalizationNode>(epsilon, conv_info, num_groups, conv_method, fast_math_hint… in fuse_convolution_with_batch_normalization()
593 const auto conv_method = conv_node->convolution_method(); in fuse_convolution_with_post_ops() local
598 …id = g.add_node<FusedConvolutionWithPostOpNode>(conv_info, num_groups, conv_method, fast_math_hint… in fuse_convolution_with_post_ops()
693 const auto conv_method = conv_node->convolution_method(); in fuse_convolution_batch_normalization_with_post_ops() local
700 …tionBatchNormalizationWithPostOpsNode>(epsilon, conv_info, num_groups, conv_method, fast_math_hint… in fuse_convolution_batch_normalization_with_post_ops()
|
| H A D | GroupedConvolutionMutator.cpp | 143 const ConvolutionMethod conv_method = conv_node->convolution_method(); in mutate() local
168 … conv_info, conv_method, fused_act_info, fast_math_hint, num_groups); in mutate()
|
| /aosp_15_r20/external/ComputeLibrary/utils/ |
| H A D | TypePrinter.h | 2285 inline ::std::ostream &operator<<(::std::ostream &os, const ConvolutionMethod &conv_method)
2287 switch(conv_method)
2317 inline std::string to_string(const ConvolutionMethod &conv_method) in to_string() argument
2320 str << conv_method; in to_string()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Convolution.cpp | 515 const Conv2dMethod conv_method) { in pack_weights() argument
527 if (conv_method == Conv2dDepthwise) { in pack_weights()
|