Home
last modified time | relevance | path

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

/aosp_15_r20/external/ComputeLibrary/src/graph/nodes/
H A DFusedDepthwiseConvolutionBatchNormalizationNode.cpp35 FusedDepthwiseConvolutionBatchNormalizationNode::FusedDepthwiseConvolutionBatchNormalizationNode(fl… in FusedDepthwiseConvolutionBatchNormalizationNode() function in arm_compute::graph::FusedDepthwiseConvolutionBatchNormalizationNode
46 void FusedDepthwiseConvolutionBatchNormalizationNode::set_depthwise_convolution_method(DepthwiseCon… in set_depthwise_convolution_method()
51 DepthwiseConvolutionMethod FusedDepthwiseConvolutionBatchNormalizationNode::depthwise_convolution_m… in depthwise_convolution_method()
56 float FusedDepthwiseConvolutionBatchNormalizationNode::epsilon() const in epsilon()
61 PadStrideInfo FusedDepthwiseConvolutionBatchNormalizationNode::convolution_info() const in convolution_info()
66 unsigned int FusedDepthwiseConvolutionBatchNormalizationNode::depth_multiplier() const in depth_multiplier()
71 ActivationLayerInfo FusedDepthwiseConvolutionBatchNormalizationNode::fused_activation() const in fused_activation()
76 void FusedDepthwiseConvolutionBatchNormalizationNode::set_fused_activation(ActivationLayerInfo fuse… in set_fused_activation()
81 TensorDescriptor FusedDepthwiseConvolutionBatchNormalizationNode::compute_output_descriptor(const T… in compute_output_descriptor()
105 bool FusedDepthwiseConvolutionBatchNormalizationNode::forward_descriptors() in forward_descriptors()
[all …]
/aosp_15_r20/external/ComputeLibrary/src/graph/
H A DDataLayerVisitor.cpp142 void DataLayerVisitor::visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) in visit()
145 add_generic_layer_data<FusedDepthwiseConvolutionBatchNormalizationNode>(_layer_data, n); in visit()
146 add_convolution_layer_data<FusedDepthwiseConvolutionBatchNormalizationNode>(_layer_data, n); in visit()
H A DINodeVisitor.cpp96 void DefaultNodeVisitor::visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) in visit()
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/nodes/
H A DFusedDepthwiseConvolutionBatchNormalizationNode.h34 class FusedDepthwiseConvolutionBatchNormalizationNode final : public INode
45 FusedDepthwiseConvolutionBatchNormalizationNode(float epsilon,
H A DNodesFwd.h53 class FusedDepthwiseConvolutionBatchNormalizationNode; variable
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/
H A DINodeVisitor.h123 virtual void visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) = 0;
220 virtual void visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) override;
H A DDataLayerVisitor.h52 void visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) override;
/aosp_15_r20/external/ComputeLibrary/src/graph/mutators/
H A DInPlaceOperationMutator.cpp114 …conv_info = polymorphic_downcast<FusedDepthwiseConvolutionBatchNormalizationNode *>(node.ge… in try_in_place_depthwiseconv()
115 …depth_multiplier = polymorphic_downcast<FusedDepthwiseConvolutionBatchNormalizationNode *>(node.ge… in try_in_place_depthwiseconv()
H A DNodeFusionMutator.cpp193 …const NodeID fused_id = g.add_node<FusedDepthwiseConvolutionBatchNormalizationNode>(epsilon, conv_… in fuse_depthwise_convolution_with_batch_normalization()
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/printers/
H A DDotGraphPrinter.h62 void visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) override;
/aosp_15_r20/external/ComputeLibrary/src/graph/printers/
H A DDotGraphPrinter.cpp104 void DotGraphVisitor::visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) in visit()
/aosp_15_r20/external/ComputeLibrary/src/graph/backends/NEON/
H A DNEFunctionFactory.cpp168 …FusedLayerTypes, NETargetInfo>(*polymorphic_downcast<FusedDepthwiseConvolutionBatchNormalizationNo… in create()
/aosp_15_r20/external/ComputeLibrary/src/graph/backends/CL/
H A DCLFunctionsFactory.cpp280 …FusedLayerTypes, CLTargetInfo>(*polymorphic_downcast<FusedDepthwiseConvolutionBatchNormalizationNo… in create()
/aosp_15_r20/external/ComputeLibrary/arm_compute/graph/backends/
H A DFunctionHelpers.h279 …used_depthwise_convolution_batch_normalization_layer(FusedDepthwiseConvolutionBatchNormalizationNo… in create_fused_depthwise_convolution_batch_normalization_layer()
/aosp_15_r20/external/ComputeLibrary/src/
H A DBUILD.bazel77 "graph/nodes/FusedDepthwiseConvolutionBatchNormalizationNode.cpp",
H A DCMakeLists.txt78 graph/nodes/FusedDepthwiseConvolutionBatchNormalizationNode.cpp