Commit bba2b3bd authored by Adam Straw's avatar Adam Straw Committed by Scott Cyphers

re-enable quantize_clamp_int32 test on CPU (#2090)

* re-enable quantize_clamp_int32 test on CPU

* MLKDNN typo
parent 1f40160d
......@@ -702,6 +702,11 @@ namespace ngraph
auto dequantize = static_cast<op::Dequantize*>(node);
auto offset_const_op =
std::static_pointer_cast<ngraph::op::Constant>(dequantize->get_argument(2));
// TODO: MKLDNN only handles float / not double
if (node->get_output_element_type(0) != element::f32)
{
return;
}
if (node->get_input_element_type(0) == element::u8)
{
auto offset = offset_const_op->get_vector<uint8_t>();
......@@ -734,6 +739,11 @@ namespace ngraph
{
return;
}
// TODO: MKLDNN only handles float / not double
if (node->get_input_element_type(0) != element::f32)
{
return;
}
if (node->get_output_element_type(0) == element::u8)
{
auto offset = offset_const_op->get_vector<uint8_t>();
......
......@@ -13,7 +13,6 @@ shape_of_vector
shape_of_matrix
shape_of_5d
sum_stable_acc_double
quantize_clamp_int32
# failing in CI build but passing on local machine
max_3d_to_scalar_int32
......
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