diff options
Diffstat (limited to 'src/graph/nodes/BatchNormalizationLayer.cpp')
| -rw-r--r-- | src/graph/nodes/BatchNormalizationLayer.cpp | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/src/graph/nodes/BatchNormalizationLayer.cpp b/src/graph/nodes/BatchNormalizationLayer.cpp index 927ce5e728..7851aa5b9e 100644 --- a/src/graph/nodes/BatchNormalizationLayer.cpp +++ b/src/graph/nodes/BatchNormalizationLayer.cpp @@ -78,6 +78,9 @@ std::unique_ptr<arm_compute::IFunction> BatchNormalizationLayer::instantiate_nod node_ctx.add_output(out); node_ctx.add_parameter<float>("epsilon", _epsilon); + // Configure operation + auto func = OperationRegistry::get().find_operation(OperationType::BatchNormalizationLayer, _target_hint)->configure(node_ctx); + // Fill tensors if(!mean_is_loaded) { @@ -97,5 +100,5 @@ std::unique_ptr<arm_compute::IFunction> BatchNormalizationLayer::instantiate_nod } // Get function - return OperationRegistry::get().find_operation(OperationType::BatchNormalizationLayer, _target_hint)->configure(node_ctx); + return func; }
\ No newline at end of file |