Commit 16973213 authored by Khalique's avatar Khalique
Browse files

update error message

parent bda63363
...@@ -366,7 +366,7 @@ struct tf_parser ...@@ -366,7 +366,7 @@ struct tf_parser
size_t input_size = args.front()->get_shape().lens().size(); size_t input_size = args.front()->get_shape().lens().size();
if(axis > input_size) if(axis > input_size)
{ {
MIGRAPHX_THROW("Error in protobuf: axis must be smaller than input size"); MIGRAPHX_THROW("TF_PARSER: axis value of " + to_string(axis) + " must be smaller than input size " + to_string(input_size));
} }
// check if input arg needs axis to be converted to NCHW // check if input arg needs axis to be converted to NCHW
if(input_size >= 4) if(input_size >= 4)
......
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