Home
last modified time | relevance | path

Searched refs:BatchNormalizationLayer (Results 1 – 25 of 45) sorted by relevance

12

/external/armnn/src/armnn/layers/
DBatchNormalizationLayer.cpp16 BatchNormalizationLayer::BatchNormalizationLayer(const armnn::BatchNormalizationDescriptor& param, … in BatchNormalizationLayer() function in armnn::BatchNormalizationLayer
21 std::unique_ptr<IWorkload> BatchNormalizationLayer::CreateWorkload(const IWorkloadFactory& factory)… in CreateWorkload()
40 BatchNormalizationLayer* BatchNormalizationLayer::Clone(Graph& graph) const in Clone()
42 auto layer = CloneBase<BatchNormalizationLayer>(graph, m_Param, GetName()); in Clone()
52 void BatchNormalizationLayer::ValidateTensorShapesFromInputs() in ValidateTensorShapesFromInputs()
68 Layer::ImmutableConstantTensors BatchNormalizationLayer::GetConstantTensorsByRef() const in GetConstantTensorsByRef()
74 void BatchNormalizationLayer::ExecuteStrategy(IStrategy& strategy) const in ExecuteStrategy()
DBatchNormalizationLayer.hpp15 class BatchNormalizationLayer : public LayerWithParameters<BatchNormalizationDescriptor> class
35 BatchNormalizationLayer* Clone(Graph& graph) const override;
48 BatchNormalizationLayer(const BatchNormalizationDescriptor& param, const char* name);
51 ~BatchNormalizationLayer() = default;
/external/ComputeLibrary/examples/
Dgraph_inception_v4.cpp87 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "/cnn_data/inceptionv4_model/Conv2d_1a_… in do_setup()
99 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "/cnn_data/inceptionv4_model/Conv2d_2a_… in do_setup()
111 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "/cnn_data/inceptionv4_model/Conv2d_2b_… in do_setup()
211 …<< BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_3x3_Ba… in get_mixed_3a()
231 …<< BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_0a_1x1_Ba… in get_mixed_4a()
242 …<< BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "Branch_0_Conv2d_1a_3x3_Ba… in get_mixed_4a()
255 …<< BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0a_1x1_Ba… in get_mixed_4a()
266 …<< BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0b_1x7_Ba… in get_mixed_4a()
277 …<< BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_0c_7x1_Ba… in get_mixed_4a()
288 …<< BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "Branch_1_Conv2d_1a_3x3_Ba… in get_mixed_4a()
[all …]
Dgraph_yolov3.cpp88 << BatchNormalizationLayer( in do_setup()
102 << BatchNormalizationLayer( in do_setup()
116 << BatchNormalizationLayer( in do_setup()
130 << BatchNormalizationLayer( in do_setup()
144 << BatchNormalizationLayer( in do_setup()
159 << BatchNormalizationLayer( in do_setup()
182 << BatchNormalizationLayer( in do_setup()
199 << BatchNormalizationLayer( in do_setup()
213 << BatchNormalizationLayer( in do_setup()
227 << BatchNormalizationLayer( in do_setup()
[all …]
Dgraph_inception_resnet_v2.cpp98 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Conv2d_1a_3x3_BatchNorm_moving_mean.np… in do_setup()
111 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Conv2d_2a_3x3_BatchNorm_moving_mean.np… in do_setup()
124 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Conv2d_2b_3x3_BatchNorm_moving_mean.np… in do_setup()
139 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Conv2d_3b_1x1_BatchNorm_moving_mean.np… in do_setup()
152 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Conv2d_4a_3x3_BatchNorm_moving_mean.np… in do_setup()
176 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Conv2d_7b_1x1_BatchNorm_moving_mean.np… in do_setup()
227 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Mixed_5b_Branch_0_Conv2d_1x1_BatchNorm… in block_mixed_5b()
242 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Mixed_5b_Branch_1_Conv2d_0a_1x1_BatchN… in block_mixed_5b()
254 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Mixed_5b_Branch_1_Conv2d_0b_5x5_BatchN… in block_mixed_5b()
269 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Mixed_5b_Branch_2_Conv2d_0a_1x1_BatchN… in block_mixed_5b()
[all …]
Dgraph_inception_v3.cpp84 << BatchNormalizationLayer(get_weights_accessor(data_path, in do_setup()
97 << BatchNormalizationLayer(get_weights_accessor(data_path, in do_setup()
111 << BatchNormalizationLayer(get_weights_accessor(data_path, in do_setup()
127 << BatchNormalizationLayer(get_weights_accessor(data_path, in do_setup()
141 << BatchNormalizationLayer(get_weights_accessor(data_path, in do_setup()
251 << BatchNormalizationLayer( in get_inception_node_A()
267 << BatchNormalizationLayer( in get_inception_node_A()
281 << BatchNormalizationLayer( in get_inception_node_A()
297 << BatchNormalizationLayer( in get_inception_node_A()
311 << BatchNormalizationLayer( in get_inception_node_A()
[all …]
Dgraph_inception_resnet_v1.cpp114 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Conv2d_1a_3x3_BatchNorm_moving_mean.np… in do_setup()
127 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Conv2d_2a_3x3_BatchNorm_moving_mean.np… in do_setup()
140 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Conv2d_2b_3x3_BatchNorm_moving_mean.np… in do_setup()
155 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Conv2d_3b_1x1_BatchNorm_moving_mean.np… in do_setup()
168 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Conv2d_4a_3x3_BatchNorm_moving_mean.np… in do_setup()
181 …<< BatchNormalizationLayer(get_weights_accessor(data_path, "Conv2d_4b_3x3_BatchNorm_moving_mean.np… in do_setup()
262 …<< BatchNormalizationLayer(get_weights_accessor(data_path, unit_path + "Branch_0_Conv2d_1x1_BatchN… in block35_repeat()
277 …<< BatchNormalizationLayer(get_weights_accessor(data_path, unit_path + "Branch_1_Conv2d_0a_1x1_Bat… in block35_repeat()
289 …<< BatchNormalizationLayer(get_weights_accessor(data_path, unit_path + "Branch_1_Conv2d_0b_3x3_Bat… in block35_repeat()
304 …<< BatchNormalizationLayer(get_weights_accessor(data_path, unit_path + "Branch_2_Conv2d_0a_1x1_Bat… in block35_repeat()
[all …]
Dgraph_resnet50.cpp86 << BatchNormalizationLayer( in do_setup()
166 << BatchNormalizationLayer( in add_residual_block()
181 << BatchNormalizationLayer( in add_residual_block()
196 << BatchNormalizationLayer( in add_residual_block()
213 << BatchNormalizationLayer( in add_residual_block()
Dgraph_resnet_v2_50.cpp96 graph << BatchNormalizationLayer( in do_setup()
165 preact << BatchNormalizationLayer( in add_residual_block()
202 << BatchNormalizationLayer( in add_residual_block()
216 << BatchNormalizationLayer( in add_residual_block()
Dgraph_shufflenet.cpp103 << BatchNormalizationLayer( in do_setup()
196 << BatchNormalizationLayer( in add_residual_block()
211 << BatchNormalizationLayer( in add_residual_block()
224 << BatchNormalizationLayer( in add_residual_block()
Dgraph_mobilenet_v2.cpp150 … << BatchNormalizationLayer(get_weights_accessor(data_path, "Conv_BatchNorm_moving_mean.npy"), in create_graph_float()
182 … << BatchNormalizationLayer(get_weights_accessor(data_path, "Conv_1_BatchNorm_moving_mean.npy"), in create_graph_float()
214 …<< BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "expand_BatchNorm_moving_m… in get_expanded_conv_float()
230 …<< BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "depthwise_BatchNorm_movin… in get_expanded_conv_float()
244 …<< BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "project_BatchNorm_moving_… in get_expanded_conv_float()
Dgraph_resnet12.cpp179 << BatchNormalizationLayer( in add_residual_block()
194 << BatchNormalizationLayer( in add_residual_block()
Dgraph_mobilenet.cpp147 << BatchNormalizationLayer( in create_graph_float()
292 << BatchNormalizationLayer( in get_dwsc_node_float()
306 << BatchNormalizationLayer( in get_dwsc_node_float()
Dgraph_ssd_mobilenet.cpp137 … << BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "dw_bn_mean.npy"), in get_node_A_float()
150 << BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "bn_mean.npy"), in get_node_A_float()
173 << BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "1_bn_mean.npy"), in get_node_B_float()
186 << BatchNormalizationLayer(get_weights_accessor(data_path, total_path + "2_bn_mean.npy"), in get_node_B_float()
241 conv_11 << BatchNormalizationLayer(get_weights_accessor(data_path, "conv0_bn_mean.npy"), in create_graph_float()
/external/armnn/src/armnnDeserializer/test/
DDeserializeBatchNormalization.cpp56 layer_type: "BatchNormalizationLayer", in BatchNormalizationFixture()
60 layerName: "BatchNormalizationLayer", in BatchNormalizationFixture()
/external/armnn/src/armnn/optimizations/
DFuseBatchNorm.hpp41 auto batchNormLayer = PolymorphicDowncast<BatchNormalizationLayer*>(&child); in Run()
221 BatchNormalizationLayer,
226 BatchNormalizationLayer,
231 BatchNormalizationLayer,
236 BatchNormalizationLayer,
/external/ComputeLibrary/arm_compute/graph/nodes/
DBatchNormalizationLayerNode.h66 static constexpr NodeType node_type = NodeType::BatchNormalizationLayer;
/external/armnn/src/backends/neon/
DNeonBackend.cpp284 BatchNormalizationLayer* baseLayer = in OptimizeSubgraphView()
285 PolymorphicDowncast<BatchNormalizationLayer*>(&base); in OptimizeSubgraphView()
299 BatchNormalizationLayer* replacementLayer = in OptimizeSubgraphView()
300 … FuseBatchNormalizationLayer<BatchNormalizationLayer>(optimizationViews, in OptimizeSubgraphView()
/external/armnn/src/backends/backendsCommon/test/
DLayerReleaseConstantDataTest.cpp33BatchNormalizationLayer* const layer = graph.AddLayer<BatchNormalizationLayer>(layerDesc, "layer");
/external/armnn/src/backends/cl/
DClBackend.cpp432 BatchNormalizationLayer* baseLayer = in OptimizeSubgraphView()
433 PolymorphicDowncast<BatchNormalizationLayer*>(&base); in OptimizeSubgraphView()
447 BatchNormalizationLayer* replacementLayer = in OptimizeSubgraphView()
448 … FuseBatchNormalizationLayer<BatchNormalizationLayer>(optimizationViews, in OptimizeSubgraphView()
/external/armnn/src/armnn/test/
DOptimizerTests.cpp890 auto batchNorm = graph.AddLayer<BatchNormalizationLayer>(batchNormDescriptor, "batchNorm");
930 &IsLayerOfType<BatchNormalizationLayer>,
940 &IsLayerOfType<BatchNormalizationLayer>,
971 auto batchNorm = graph.AddLayer<BatchNormalizationLayer>(batchNormDescriptor, "batchNorm");
985 &IsLayerOfType<armnn::BatchNormalizationLayer>,
995 &IsLayerOfType<armnn::BatchNormalizationLayer>,
/external/ComputeLibrary/arm_compute/graph/
DTypePrinter.h71 case NodeType::BatchNormalizationLayer:
DTypes.h203 BatchNormalizationLayer, enumerator
/external/ComputeLibrary/src/graph/mutators/
DSyntheticDataTypeMutator.cpp86 const std::set<NodeType> optimized_node_types = { NodeType::BatchNormalizationLayer }; in remove_optimized_nodes()
DInPlaceOperationMutator.cpp219 NodeType::BatchNormalizationLayer, in mutate()

12