TensorFlow.NET/test/TensorFlowNET.UnitTest/ConsumersTest.cs at master · zews/TensorFlow.NET · GitHub

32 lines (28 loc) · 848 Bytes

using Microsoft.VisualStudio.TestTools.UnitTesting;

using Tensorflow;

using static Tensorflow.Binding;

namespace TensorFlowNET.UnitTest

{

[TestClass]

public class ConsumersTest : CApiTest

{

[TestMethod]

public void Constant()

{

var X = tf.placeholder(tf.float64);

var W = tf.constant(1.0D);

var mul = tf.multiply(X, W);

EXPECT_EQ(1, X.op.OutputNumConsumers(0));

EXPECT_EQ(1, W.op.OutputNumConsumers(0));

}

[TestMethod]

public void Variable()

{

var X = tf.placeholder(tf.float64);

var W = tf.Variable(1.0D, name: "var");

var mul = tf.multiply(X, W);

EXPECT_EQ(1, X.op.OutputNumConsumers(0));

//EXPECT_EQ(1, W.op.OutputNumConsumers(0));

}

Expand file tree