Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/src/graph/nodes/
H A DFusedConvolutionBatchNormalizationNode.cpp35 FusedConvolutionBatchNormalizationNode::FusedConvolutionBatchNormalizationNode(float epsilon, PadSt… in FusedConvolutionBatchNormalizationNode() function in arm_compute::graph::FusedConvolutionBatchNormalizationNode
46 void FusedConvolutionBatchNormalizationNode::set_convolution_method(ConvolutionMethod method) in set_convolution_method()
51 float FusedConvolutionBatchNormalizationNode::epsilon() const in epsilon()
56 ConvolutionMethod FusedConvolutionBatchNormalizationNode::convolution_method() const in convolution_method()
61 void FusedConvolutionBatchNormalizationNode::set_fast_math_hint(FastMathHint hint) in set_fast_math_hint()
66 FastMathHint FusedConvolutionBatchNormalizationNode::fast_math_hint() const in fast_math_hint()
71 PadStrideInfo FusedConvolutionBatchNormalizationNode::convolution_info() const in convolution_info()
76 unsigned int FusedConvolutionBatchNormalizationNode::num_groups() const in num_groups()
81 ActivationLayerInfo FusedConvolutionBatchNormalizationNode::fused_activation() const in fused_activation()
86 void FusedConvolutionBatchNormalizationNode::set_fused_activation(ActivationLayerInfo fused_activat… in set_fused_activation()
[all …]
/aosp_15_r20/external/ComputeLibrary/src/graph/
H A DDataLayerVisitor.cpp126 void DataLayerVisitor::visit(FusedConvolutionBatchNormalizationNode &n) in visit()
129 add_generic_layer_data<FusedConvolutionBatchNormalizationNode>(_layer_data, n); in visit()
130 add_convolution_layer_data<FusedConvolutionBatchNormalizationNode>(_layer_data, n); in visit()
131 add_convolution_layer_method<FusedConvolutionBatchNormalizationNode>(_layer_data, n); in visit()
H A DINodeVisitor.cpp84 void DefaultNodeVisitor::visit(FusedConvolutionBatchNormalizationNode &n) in visit()
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/nodes/
H A DFusedConvolutionBatchNormalizationNode.h34 class FusedConvolutionBatchNormalizationNode final : public INode
46 FusedConvolutionBatchNormalizationNode(float epsilon, PadStrideInfo info,
H A DNodesFwd.h51 class FusedConvolutionBatchNormalizationNode; variable
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/
H A DINodeVisitor.h108 virtual void visit(FusedConvolutionBatchNormalizationNode &n) = 0;
217 virtual void visit(FusedConvolutionBatchNormalizationNode &n) override;
H A DDataLayerVisitor.h50 void visit(FusedConvolutionBatchNormalizationNode &n) override;
/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DNodeFusionMutator.cpp119 …const NodeID fused_id = g.add_node<FusedConvolutionBatchNormalizationNode>(epsilon, conv_info, num… in fuse_convolution_with_batch_normalization()
644 …nv_node = arm_compute::utils::cast::polymorphic_downcast<FusedConvolutionBatchNormalizationNode *>… in fuse_convolution_batch_normalization_with_post_ops()
853 …detail::fuse_layer<FusedConvolutionBatchNormalizationNode>(g, cl_target_prec, detail::fuse_convolu… in mutate()
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/printers/
H A DDotGraphPrinter.h59 void visit(FusedConvolutionBatchNormalizationNode &n) override;
/aosp_15_r20/external/ComputeLibrary/src/graph/printers/
H A DDotGraphPrinter.cpp80 void DotGraphVisitor::visit(FusedConvolutionBatchNormalizationNode &n) in visit()
/aosp_15_r20/external/ComputeLibrary/src/graph/backends/NEON/
H A DNEFunctionFactory.cpp166 …r<NEFusedLayerTypes, NETargetInfo>(*polymorphic_downcast<FusedConvolutionBatchNormalizationNode *>… in create()
/aosp_15_r20/external/ComputeLibrary/src/graph/backends/CL/
H A DCLFunctionsFactory.cpp276 …r<CLFusedLayerTypes, CLTargetInfo>(*polymorphic_downcast<FusedConvolutionBatchNormalizationNode *>… in create()
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/backends/
H A DFunctionHelpers.h222 …tion> create_fused_convolution_batch_normalization_layer(FusedConvolutionBatchNormalizationNode &n… in create_fused_convolution_batch_normalization_layer()
/aosp_15_r20/external/ComputeLibrary/src/
H A DBUILD.bazel74 "graph/nodes/FusedConvolutionBatchNormalizationNode.cpp",
H A DCMakeLists.txt75 graph/nodes/FusedConvolutionBatchNormalizationNode.cpp