[test] modify model supporting part of low_level_zero plugin (including correspoding docs)

This commit is contained in:
Zhongkai Zhao
2023-10-05 15:10:31 +08:00
parent d1fcc0fa4d
commit db40e086c8
3 changed files with 0 additions and 22 deletions

View File

@@ -53,16 +53,6 @@ def output_transform_fn(x):
return dict(output=x)
def output_transform_fn(x):
if isinstance(x, KeyedTensor):
output = dict()
for key in x.keys():
output[key] = x[key]
return output
else:
return dict(output=x)
def get_ebc():
# EmbeddingBagCollection
eb1_config = EmbeddingBagConfig(name="t1", embedding_dim=SHAPE, num_embeddings=SHAPE, feature_names=["f1"])