Searched refs:depth_conv_node (Results 1 – 1 of 1) sorted by relevance
166 …auto *depth_conv_node = arm_compute::utils::cast::polymorphic_downcast<DepthwiseConvolutionLayerNo… in fuse_depthwise_convolution_with_batch_normalization() local173 if(depth_conv_node->output(0)->accessor() == nullptr) in fuse_depthwise_convolution_with_batch_normalization()175 const Target assigned_target = depth_conv_node->assigned_target(); in fuse_depthwise_convolution_with_batch_normalization()178 const auto depth_conv_input_id = depth_conv_node->input_edge(0)->producer_id(); in fuse_depthwise_convolution_with_batch_normalization()179 const auto conv_weights_id = depth_conv_node->input_edge(1)->producer_id(); in fuse_depthwise_convolution_with_batch_normalization()180 const auto conv_info = depth_conv_node->convolution_info(); in fuse_depthwise_convolution_with_batch_normalization()181 const auto depth_conv_method = depth_conv_node->depthwise_convolution_method(); in fuse_depthwise_convolution_with_batch_normalization()182 const auto depth_multiplier = depth_conv_node->depth_multiplier(); in fuse_depthwise_convolution_with_batch_normalization()195 if(depth_conv_node->input_edge(2) != nullptr) in fuse_depthwise_convolution_with_batch_normalization()197 const auto conv_bias_id = depth_conv_node->input_edge(2)->producer_id(); in fuse_depthwise_convolution_with_batch_normalization()[all …]