IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix DynamicStitch tests (#14005)

Browse Source
This commit is contained in:
Ivan Tikhonov
2022-11-16 01:24:38 +04:00
committed by GitHub
parent 57293a6833
commit 8d3b382ca9
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/layer_tests/tensorflow_tests/test_tf_ParallelDynamicStitch.py
View File

@@ -19,7 +19,10 @@ class TestParallelDynamicStitch(CommonTFLayerTest):
indices_shape = inputs_info[indices_in_name]
num_elements = num_elements + np.prod(indices_shape, dtype=int)
indices_array = np.arange(np.random.randint(1, num_elements+1), dtype=np.intc)
# we support DynamicStitch via decomposition to subgraph with ScatterUpdate op
# ScatterUpdate has undefined behavior if there are multiple identical indexes
# indices_array = np.arange(np.random.randint(1, num_elements+1), dtype=np.intc)
indices_array = np.arange(num_elements, dtype=np.intc)
np.random.shuffle(indices_array)
indices_array = np.resize(indices_array, num_elements)