Searched refs:FusedDepthwiseConvolutionBatchNormalizationNode (Results 1 – 16 of 16) sorted by relevance
35 FusedDepthwiseConvolutionBatchNormalizationNode::FusedDepthwiseConvolutionBatchNormalizationNode(fl… in FusedDepthwiseConvolutionBatchNormalizationNode() function in arm_compute::graph::FusedDepthwiseConvolutionBatchNormalizationNode46 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 …]
142 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()
96 void DefaultNodeVisitor::visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) in visit()
34 class FusedDepthwiseConvolutionBatchNormalizationNode final : public INode45 FusedDepthwiseConvolutionBatchNormalizationNode(float epsilon,
53 class FusedDepthwiseConvolutionBatchNormalizationNode; variable
123 virtual void visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) = 0;220 virtual void visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) override;
52 void visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) override;
114 …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()
193 …const NodeID fused_id = g.add_node<FusedDepthwiseConvolutionBatchNormalizationNode>(epsilon, conv_… in fuse_depthwise_convolution_with_batch_normalization()
62 void visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) override;
104 void DotGraphVisitor::visit(FusedDepthwiseConvolutionBatchNormalizationNode &n) in visit()
168 …FusedLayerTypes, NETargetInfo>(*polymorphic_downcast<FusedDepthwiseConvolutionBatchNormalizationNo… in create()
280 …FusedLayerTypes, CLTargetInfo>(*polymorphic_downcast<FusedDepthwiseConvolutionBatchNormalizationNo… in create()
279 …used_depthwise_convolution_batch_normalization_layer(FusedDepthwiseConvolutionBatchNormalizationNo… in create_fused_depthwise_convolution_batch_normalization_layer()
77 "graph/nodes/FusedDepthwiseConvolutionBatchNormalizationNode.cpp",
78 graph/nodes/FusedDepthwiseConvolutionBatchNormalizationNode.cpp