Searched refs:B_sizes (Results 1 – 8 of 8) sorted by relevance
26 const std::vector<Dimension>& B_sizes = inputs[1]->sizes(); in adapt_broadcast_forward_compatibility() local29 if (node->i(kaxis) != (int) (A_sizes.size() - B_sizes.size())) { in adapt_broadcast_forward_compatibility()34 std::vector<Dimension> new_sizes = B_sizes; in adapt_broadcast_forward_compatibility()35 auto size = A_sizes.size() > B_sizes.size() ? A_sizes.size() - B_sizes.size() : 0; in adapt_broadcast_forward_compatibility()39 axes.emplace_back(B_sizes.size() + i); in adapt_broadcast_forward_compatibility()
23 std::vector<Dimension> B_sizes = inputs[i]->sizes(); in adapt_sum_8_7() local24 assert_numpy_multibroadcastable(A_sizes, B_sizes); in adapt_sum_8_7()
26 const std::vector<Dimension>& B_sizes = inputs[1]->sizes(); in adapt_broadcast_backward_compatibility() local31 int req_broadcast = check_numpy_unibroadcastable_and_require_broadcast(A_sizes, B_sizes); in adapt_broadcast_backward_compatibility()
22 const std::vector<Dimension>& B_sizes = inputs[1]->sizes(); in adapt_broadcast_forward_compatibility() local25 if (node->i(kaxis) != (int) (A_sizes.size() - B_sizes.size())) { in adapt_broadcast_forward_compatibility()30 std::vector<Dimension> new_sizes = B_sizes; in adapt_broadcast_forward_compatibility()31 for (int i = 0; i < (int) (A_sizes.size() - B_sizes.size()); i++) { in adapt_broadcast_forward_compatibility()32 axes.emplace_back(B_sizes.size() + i); in adapt_broadcast_forward_compatibility()
19 std::vector<Dimension> B_sizes = inputs[i]->sizes(); in adapt_sum_8_7() local20 assert_numpy_multibroadcastable(A_sizes, B_sizes); in adapt_sum_8_7()
22 const std::vector<Dimension>& B_sizes = inputs[1]->sizes(); in adapt_broadcast_backward_compatibility() local27 int req_broadcast = check_numpy_unibroadcastable_and_require_broadcast(A_sizes, B_sizes); in adapt_broadcast_backward_compatibility()
43 const std::vector<Dimension>& B_sizes = *B_ptr; in assert_numpy_multibroadcastable() local44 int axis = (int) (A_sizes.size() - B_sizes.size()); in assert_numpy_multibroadcastable()45 for (int i = 0; i < (int) B_sizes.size(); i++) { in assert_numpy_multibroadcastable()46 ONNX_ASSERTM(B_sizes[i].dim == A_sizes[axis + i].dim || in assert_numpy_multibroadcastable()47 … B_sizes[i].dim == 1 || A_sizes[axis + i].dim == 1, "Dimension %d of input %d does not match " in assert_numpy_multibroadcastable()
47 const std::vector<Dimension>& B_sizes = *B_ptr; in assert_numpy_multibroadcastable() local48 int axis = (int) (A_sizes.size() - B_sizes.size()); in assert_numpy_multibroadcastable()49 for (int i = 0; i < (int) B_sizes.size(); i++) { in assert_numpy_multibroadcastable()50 ONNX_ASSERTM(B_sizes[i].dim == A_sizes[axis + i].dim || in assert_numpy_multibroadcastable()51 … B_sizes[i].dim == 1 || A_sizes[axis + i].dim == 1, "Dimension %d of input %d does not match " in assert_numpy_multibroadcastable()