ir_version: 4 producer_name: "nGraph ONNX Importer" graph { node { input: "x" input: "k" output: "values" output: "indices" op_type: "TopK" } name: "test_top_k" input { name: "x" type { tensor_type { elem_type: 1 shape { dim { dim_value: 3 } dim { dim_value: 4 } } } } } input { name: "k" type { tensor_type { elem_type: 7 shape { dim { dim_value: 1 } } } } } initializer { dims: 1 data_type: 7 int64_data: 3 name: "k" } output { name: "values" type { tensor_type { elem_type: 1 shape { dim { dim_value: 3 } dim { dim_value: 3 } } } } } output { name: "indices" type { tensor_type { elem_type: 7 shape { dim { dim_value: 3 } dim { dim_value: 3 } } } } } } opset_import { version: 10 }