diff --git a/test/TensorFlowNET.UnitTest/OperationsTest.cs b/test/TensorFlowNET.UnitTest/OperationsTest.cs index 28abce32..729a129d 100644 --- a/test/TensorFlowNET.UnitTest/OperationsTest.cs +++ b/test/TensorFlowNET.UnitTest/OperationsTest.cs @@ -1,6 +1,7 @@ using Microsoft.VisualStudio.TestTools.UnitTesting; using System; using System.Collections.Generic; +using System.Linq; using System.Text; using Tensorflow; using Buffer = Tensorflow.Buffer; @@ -20,6 +21,13 @@ namespace TensorFlowNET.UnitTest var handle = c_api.TF_GetAllOpList(); var buffer = new Buffer(handle); var op_list = OpList.Parser.ParseFrom(buffer); + + var _registered_ops = new Dictionary(); + foreach (var op_def in op_list.Op) + _registered_ops[op_def.Name] = op_def; + + // r1.14 added NN op + var op = _registered_ops.FirstOrDefault(x => x.Key == "NearestNeighbors"); Assert.IsTrue(op_list.Op.Count > 1000); }