Commit cb6deeb2 authored by Khalique's avatar Khalique
Browse files

formatting

parent 21bb2293
...@@ -568,19 +568,19 @@ struct tf_parser ...@@ -568,19 +568,19 @@ struct tf_parser
parse_onehot(const std::string&, attribute_map attributes, std::vector<instruction_ref> args) parse_onehot(const std::string&, attribute_map attributes, std::vector<instruction_ref> args)
{ {
// auto indices = args[0]->eval().get<int32_t>().to_vector(); // auto indices = args[0]->eval().get<int32_t>().to_vector();
size_t depth = static_cast<size_t>(args[1]->eval().at<int32_t>()); size_t depth = static_cast<size_t>(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>();
std::vector<float> depth_input(depth*depth, off_value); std::vector<float> depth_input(depth * depth, off_value);
for(int i = 0; i < depth; i++) for(int i = 0; i < depth; i++)
{ {
depth_input[depth*i + i] = on_value; depth_input[depth * i + i] = on_value;
} }
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