Commit ff009f50 authored by Khalique's avatar Khalique
Browse files

clarified reorder_data use

parent 395ccd2c
...@@ -50,6 +50,9 @@ struct tf_parser ...@@ -50,6 +50,9 @@ struct tf_parser
return axes; return axes;
} }
// tf stores certain attributes such as strides, dilations, as a 4D input.
// The first and last dims are equal to 1, and the relevant data is in dims 2 and 3.
// This helper function reorders the data to store for the respective operator member variables.
template <class T> template <class T>
void reorder_data(std::vector<T>& prev_data) const void reorder_data(std::vector<T>& prev_data) const
{ {
......
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