parent
bb6a3251a8
commit
7338257e00
@ -497,7 +497,8 @@ public:
|
||||
// Compute brodcasted dims
|
||||
input_shape = m_input.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;
|
||||
for (size_t i = 0; i < input_shape.size(); ++i) {
|
||||
const int64_t shifted_elem = i + weights_shape_size_diff;
|
||||
|
@ -59,7 +59,7 @@ ov::pass::BroadcastConstRangeReplacement::BroadcastConstRangeReplacement() {
|
||||
const auto target_dim_index =
|
||||
std::distance(const_node_shape.cbegin(),
|
||||
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;
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user