Skip to content

GitLab

Commit f8ec4fa7 authored by Khalique's avatar Khalique
Browse files

formatting

parent 51597ed7
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
src/tf/tf.cpp
View file @ f8ec4fa7
...@@ -79,7 +79,8 @@ struct tf_parser ...@@ -79,7 +79,8 @@ struct tf_parser
return result; return result;
} }
std::vector<size_t> parse_axes(const attribute_map& attributes, const std::string& s, const size_t& num_dims) const std::vector<size_t>
parse_axes(const attribute_map& attributes, const std::string& s, const size_t& num_dims) const
{ {
auto attrs = attributes.at(s).list().i(); auto attrs = attributes.at(s).list().i();
std::vector<size_t> axes; std::vector<size_t> axes;
...@@ -716,7 +717,7 @@ struct tf_parser ...@@ -716,7 +717,7 @@ struct tf_parser
{ {
op::squeeze op; op::squeeze op;
auto input_dims = args[0]->get_shape().lens(); auto input_dims = args[0]->get_shape().lens();
auto axes = attributes.at("squeeze_dims").list().i(); auto axes = attributes.at("squeeze_dims").list().i();
copy(axes, std::back_inserter(op.axes)); copy(axes, std::back_inserter(op.axes));
if(op.axes.empty()) // no squeeze_dims provided, remove any dim that equals 1 if(op.axes.empty()) // no squeeze_dims provided, remove any dim that equals 1
......
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