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

formatting

parent c5c9f667
......@@ -481,14 +481,18 @@ struct tf_parser
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;
auto begin = args[1]->eval().get<int64_t>().to_vector();;
auto end = args[2]->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();
;
op.starts = begin;
op.ends = end;
op.ends = end;
int shrink_axis_mask = 0;
std::vector<int64_t> squeeze_axes;
......@@ -498,9 +502,9 @@ struct tf_parser
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);
}
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