parent
ab879f143c
commit
27210b6505
@ -914,6 +914,7 @@ public:
|
|||||||
if (auto input_mask = getMask(m_input)) {
|
if (auto input_mask = getMask(m_input)) {
|
||||||
auto output_mask = std::make_shared<Mask>(m_output.get_partial_shape().rank().get_length());
|
auto output_mask = std::make_shared<Mask>(m_output.get_partial_shape().rank().get_length());
|
||||||
const auto constant = std::dynamic_pointer_cast<opset10::Constant>(m_weights.get_node_shared_ptr());
|
const auto constant = std::dynamic_pointer_cast<opset10::Constant>(m_weights.get_node_shared_ptr());
|
||||||
|
OPENVINO_ASSERT(!!constant, "Dynamic cast returned a nullptr");
|
||||||
const auto reduce_dims = constant->cast_vector<int64_t>();
|
const auto reduce_dims = constant->cast_vector<int64_t>();
|
||||||
|
|
||||||
auto input_mask_row = input_mask.get();
|
auto input_mask_row = input_mask.get();
|
||||||
|
Loading…
Reference in New Issue
Block a user