Fix check repeats in values in Tile (#20654)
- no action if any of repeats is zero Co-authored-by: Michal Lukaszewski <michal.lukaszewski@intel.com>
This commit is contained in:
parent
6fa4f9fd78
commit
84732515b4
@ -26,7 +26,7 @@ void tile(const char* arg,
|
||||
const Shape& out_shape,
|
||||
const size_t elem_size,
|
||||
const std::vector<int64_t>& repeats) {
|
||||
if (std::all_of(repeats.begin(), repeats.end(), [](int64_t repeat) {
|
||||
if (std::any_of(repeats.begin(), repeats.end(), [](int64_t repeat) {
|
||||
return repeat == 0;
|
||||
})) {
|
||||
return;
|
||||
|
@ -102,6 +102,14 @@ std::vector<TileParams> generateParams() {
|
||||
reference_tests::Tensor(ET_INT, {2}, std::vector<T_INT>{2, 1}),
|
||||
reference_tests::Tensor(ET, {2, 2, 3}, std::vector<T>{1, 2, 3, 1, 2, 3, 4, 5, 6, 4, 5, 6}),
|
||||
"tile_3d_to_3d_repeats_broadcast"),
|
||||
TileParams(reference_tests::Tensor(ET, {1}, std::vector<T>{1}),
|
||||
reference_tests::Tensor(ET_INT, {3}, std::vector<T_INT>{0, 2, 3}),
|
||||
reference_tests::Tensor(ET, {0}, std::vector<T>{}),
|
||||
"tile_1d_to_3d_with_zero_on_axis_0"),
|
||||
TileParams(reference_tests::Tensor(ET, {3}, std::vector<T>{1, 2, 3}),
|
||||
reference_tests::Tensor(ET_INT, {3}, std::vector<T_INT>{2, 0, 3}),
|
||||
reference_tests::Tensor(ET, {0}, std::vector<T>{}),
|
||||
"tile_1d_to_3d_with_zero_on_axis_1"),
|
||||
};
|
||||
return params;
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user