Skip to content

Commit

Permalink
[LPT] code style fix #2
Browse files Browse the repository at this point in the history
  • Loading branch information
vzinovie committed Jul 15, 2020
1 parent 62f9835 commit 2abc81b
Showing 1 changed file with 0 additions and 7 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -54,24 +54,17 @@ class AddTransformation : public LayerTransformation, public testing::WithParamI
testParams.transformationParams,
testParams.actual);

ngraph::pass::VisualizeTree("C:\\Users\\vzinovie\\Documents\\model.original").run_on_module(std::vector<std::shared_ptr<ngraph::Function> > { actualFunction });

SimpleLowPrecisionTransformer transform;
transform.add<ngraph::pass::low_precision::AddTransformation, ngraph::opset1::Add>(
low_precision::LayerTransformation::Params(testParams.transformationParams));
transform.transform(actualFunction);

ngraph::pass::VisualizeTree("C:\\Users\\vzinovie\\Documents\\model.transformed").run_on_module(std::vector<std::shared_ptr<ngraph::Function> > { actualFunction });

referenceFunction = AddFunction::getReference(
precision,
shape,
broadcast,
testParams.transformationParams,
testParams.expected);

ngraph::pass::VisualizeTree("C:\\Users\\vzinovie\\Documents\\model.reference").run_on_module(std::vector<std::shared_ptr<ngraph::Function> > { referenceFunction });

}

static std::string getTestCaseName(testing::TestParamInfo<AddTransformationParams> obj) {
Expand Down

0 comments on commit 2abc81b

Please sign in to comment.