Commit 39496181 authored by Khalique's avatar Khalique
Browse files

formatting

parent 634ea0f2
...@@ -136,9 +136,10 @@ struct tf_parser ...@@ -136,9 +136,10 @@ struct tf_parser
instruction_ref instruction_ref
parse_batchnorm(const std::string&, attribute_map attributes, std::vector<instruction_ref> args) parse_batchnorm(const std::string&, attribute_map attributes, std::vector<instruction_ref> args)
{ {
float epsilon = 1e-4f; float epsilon = 1e-4f;
float momentum = 1.f; float momentum = 1.f;
op::batch_norm_inference::bn_infer_mode_t bn_mode = op::batch_norm_inference::per_activation; op::batch_norm_inference::bn_infer_mode_t bn_mode =
op::batch_norm_inference::per_activation;
if(contains(attributes, "epsilon")) if(contains(attributes, "epsilon"))
{ {
epsilon = attributes.at("epsilon").f(); epsilon = attributes.at("epsilon").f();
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment