Skip to content

N
ngraph
Commit 351917d5 authored by nishant.b.patel's avatar nishant.b.patel
Browse files
Options

Address feedback

parent d394d986
Show whitespace changes
Inline Side-by-side
Showing with 14 additions and 7 deletions
src/ngraph/op/quantized_convolution.cpp
View file @ 351917d5
......@@ -77,20 +77,25 @@ void op::QuantizedConvolution::validate_and_infer_types()
NODE_VALIDATION_CHECK(this,
get_input_element_type(INPUT_SCALE).is_real() ||
get_input_element_type(INPUT_SCALE).is_dynamic() ||
get_input_element_type(FILTER_SCALE).is_real() ||
get_input_element_type(OUTPUT_SCALE).is_real(),
get_input_element_type(FILTER_SCALE).is_dynamic() ||
get_input_element_type(OUTPUT_SCALE).is_real() ||
get_input_element_type(OUTPUT_SCALE).is_dynamic(),
"Scale must be a floating point number");
NODE_VALIDATION_CHECK(this,
get_input_element_type(0) == get_input_element_type(INPUT_ZERO_POINT),
NODE_VALIDATION_CHECK(
this,
get_input_element_type(0).compatible(get_input_element_type(INPUT_ZERO_POINT)),
"Input Zero point element type (",
get_input_element_type(INPUT_ZERO_POINT),
") must match input element type (",
get_input_element_type(0),
")");
NODE_VALIDATION_CHECK(this,
get_input_element_type(1) == get_input_element_type(FILTER_ZERO_POINT),
NODE_VALIDATION_CHECK(
this,
get_input_element_type(1).compatible(get_input_element_type(FILTER_ZERO_POINT)),
"Filter Zero point element type (",
get_input_element_type(FILTER_ZERO_POINT),
") must match filter element type (",
......@@ -158,8 +163,10 @@ void op::QuantizedConvolution::validate_and_infer_types()
0, /* batch_axis_result, */
1 /* output_channel_axis_result, */
));
NODE_VALIDATION_CHECK(this,
get_output_element_type(0) == get_input_element_type(OUTPUT_ZERO_POINT),
NODE_VALIDATION_CHECK(
this,
get_output_element_type(0).compatible(get_input_element_type(OUTPUT_ZERO_POINT)),
"Output Zero point element type (",
get_input_element_type(OUTPUT_ZERO_POINT),
") must match output element type (",
......
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