Commit a357cb7e authored by Khalique's avatar Khalique
Browse files

formatting

parent 1e56489a
...@@ -573,17 +573,17 @@ struct tf_parser ...@@ -573,17 +573,17 @@ struct tf_parser
size_t num_indices = indices.size(); size_t num_indices = indices.size();
float on_value = args[2]->eval().at<float>(); float on_value = args[2]->eval().at<float>();
float off_value = args[3]->eval().at<float>(); float off_value = args[3]->eval().at<float>();
if (contains(attributes, "axis")) if(contains(attributes, "axis"))
axis = attributes.at("axis").i(); axis = attributes.at("axis").i();
if(axis == -1) if(axis == -1)
{ {
shape s{shape::float_type, {num_indices, static_cast<size_t>(depth)}}; shape s{shape::float_type, {num_indices, static_cast<size_t>(depth)}};
std::vector<float> output(num_indices * depth); std::vector<float> output(num_indices * depth);
std::fill(output.begin(), output.end(), off_value); std::fill(output.begin(), output.end(), off_value);
for (size_t i = 0; i < num_indices; i++) for(size_t i = 0; i < num_indices; i++)
{ {
if(indices[i] >= 0 and indices[i] < num_indices) if(indices[i] >= 0 and indices[i] < num_indices)
output[depth*i + indices[i]] = on_value; output[depth * i + indices[i]] = on_value;
} }
return prog.add_literal(s, output); return prog.add_literal(s, output);
} }
......
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