parent
bb6a3251a8
commit
7338257e00
@ -497,7 +497,8 @@ public:
|
|||||||
// Compute brodcasted dims
|
// Compute brodcasted dims
|
||||||
input_shape = m_input.get_shape();
|
input_shape = m_input.get_shape();
|
||||||
weights_shape = m_weights.get_shape();
|
weights_shape = m_weights.get_shape();
|
||||||
const int64_t input_shape_size_diff = input_shape.size() - weights_shape.size();
|
const int64_t input_shape_size_diff =
|
||||||
|
static_cast<int64_t>(input_shape.size()) - static_cast<int64_t>(weights_shape.size());
|
||||||
const int64_t weights_shape_size_diff = -input_shape_size_diff;
|
const int64_t weights_shape_size_diff = -input_shape_size_diff;
|
||||||
for (size_t i = 0; i < input_shape.size(); ++i) {
|
for (size_t i = 0; i < input_shape.size(); ++i) {
|
||||||
const int64_t shifted_elem = i + weights_shape_size_diff;
|
const int64_t shifted_elem = i + weights_shape_size_diff;
|
||||||
|
@ -59,7 +59,7 @@ ov::pass::BroadcastConstRangeReplacement::BroadcastConstRangeReplacement() {
|
|||||||
const auto target_dim_index =
|
const auto target_dim_index =
|
||||||
std::distance(const_node_shape.cbegin(),
|
std::distance(const_node_shape.cbegin(),
|
||||||
std::find(const_node_shape.cbegin(), const_node_shape.cend(), elem_count));
|
std::find(const_node_shape.cbegin(), const_node_shape.cend(), elem_count));
|
||||||
const int64_t target_dim_neg_index = target_dim_index - const_rank;
|
const int64_t target_dim_neg_index = target_dim_index - static_cast<int64_t>(const_rank);
|
||||||
|
|
||||||
NodeRegistry node_registry;
|
NodeRegistry node_registry;
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user