Commit b9e0e0fc authored by Khalique's avatar Khalique
Browse files

formatting

parent 30f03168
...@@ -569,11 +569,11 @@ struct tf_parser ...@@ -569,11 +569,11 @@ struct tf_parser
{ {
auto indices = args[0]->eval().get<int32_t>().to_vector(); auto indices = args[0]->eval().get<int32_t>().to_vector();
int depth = args[1]->eval().at<int32_t>(); int depth = args[1]->eval().at<int32_t>();
int64_t axis = -1; int64_t axis = -1;
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)
{ {
......
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