@@ -156,7 +156,7 @@ class KVCacheTests: public ::testing::Test {
156
156
std::map<ov::Output<ov::Node>, ov::PartialShape> shapes = {
157
157
{ref_model->input (0 ), kv_cache_copy.get_shape ()},
158
158
{ref_model->input (1 ), new_token_data.get_shape ()},
159
- {ref_model->input (2 ), matmul_data.get_shape ()},
159
+ {ref_model->input (2 ), matmul_data.get_shape ()}
160
160
};
161
161
ref_model->reshape (shapes);
162
162
@@ -342,8 +342,8 @@ class KVCacheTests: public ::testing::Test {
342
342
auto input2 = ref_model->get_parameters ().at (2 );
343
343
auto input3 = fuse_cache_reorder ? ref_model->get_parameters ().at (3 ) : nullptr ;
344
344
std::map<ov::Output<ov::Node>, ov::PartialShape> input_shapes = {
345
- {input0, kv_cache.get_shape ()},
346
- {input1, new_token_data.get_shape ()},
345
+ {input0, kv_cache.get_shape ()},
346
+ {input1, new_token_data.get_shape ()},
347
347
{input2, matmul_data.get_shape ()}
348
348
};
349
349
std::map<std::shared_ptr<ov::Node>, ov::Tensor> inputs = {
0 commit comments