Commit 9487dd3a authored by Khalique's avatar Khalique
Browse files

formatting

parent c5c9f667
...@@ -481,14 +481,18 @@ struct tf_parser ...@@ -481,14 +481,18 @@ struct tf_parser
return prog.add_instruction(op, args[0]); return prog.add_instruction(op, args[0]);
} }
instruction_ref parse_stridedslice(const std::string&, const attribute_map& attributes, std::vector<instruction_ref> args) instruction_ref parse_stridedslice(const std::string&,
const attribute_map& attributes,
std::vector<instruction_ref> args)
{ {
op::slice op; op::slice op;
auto begin = args[1]->eval().get<int64_t>().to_vector();; auto begin = args[1]->eval().get<int64_t>().to_vector();
auto end = args[2]->eval().get<int64_t>().to_vector();; ;
auto end = args[2]->eval().get<int64_t>().to_vector();
;
op.starts = begin; op.starts = begin;
op.ends = end; op.ends = end;
int shrink_axis_mask = 0; int shrink_axis_mask = 0;
std::vector<int64_t> squeeze_axes; std::vector<int64_t> squeeze_axes;
...@@ -498,9 +502,9 @@ struct tf_parser ...@@ -498,9 +502,9 @@ struct tf_parser
size_t num_axes = args[0]->get_shape().lens().size(); size_t num_axes = args[0]->get_shape().lens().size();
for (size_t i = 0; i < num_axes; i++) for(size_t i = 0; i < num_axes; i++)
{ {
if ((shrink_axis_mask >> i) & 1) if((shrink_axis_mask >> i) & 1)
squeeze_axes.push_back(i); squeeze_axes.push_back(i);
} }
auto l0 = prog.add_instruction(op, args[0]); auto l0 = prog.add_instruction(op, args[0]);
......
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