Commit 867a8b67 authored by Gleb Kazantaev's avatar Gleb Kazantaev Committed by Scott Cyphers

Convert slope to data type (#3984)

* Convert slope to data type

* UPdated Mod decomposition to use V1 Subtract
parent 7b9f581a
......@@ -49,7 +49,7 @@ NodeVector op::v1::Mod::decompose_op() const
// truncated(a / b) * b
const auto multiplication = make_shared<op::v1::Multiply>(division, divisor, m_auto_broadcast);
// a mod b = a - truncated(a / b) * b
const auto mod = make_shared<op::Subtract>(dividend, multiplication, m_auto_broadcast);
const auto mod = make_shared<op::v1::Subtract>(dividend, multiplication, m_auto_broadcast);
// apply sign of dividend
return {make_shared<op::v1::Multiply>(dividend_sign, mod, m_auto_broadcast)};
......
......@@ -40,6 +40,7 @@ NodeVector op::PRelu::decompose_op() const
auto data = input_value(0);
auto data_shape = data.get_shape();
auto slope = input_value(1);
slope = std::make_shared<op::Convert>(slope, data.get_element_type());
auto slope_shape = slope.get_shape();
if ((slope_shape.size() == 1) && (slope_shape.at(0) != 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