From c2c2ac63e7c13f1ab009749fc5d64a9baa3a860d Mon Sep 17 00:00:00 2001 From: haiping008 Date: Tue, 23 Apr 2019 17:01:26 -0500 Subject: [PATCH] check NearestNeighbors op --- test/TensorFlowNET.UnitTest/OperationsTest.cs | 8 ++++++++ 1 file changed, 8 insertions(+) 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); }