|
|
|
@@ -0,0 +1,34 @@ |
|
|
|
using Microsoft.VisualStudio.TestTools.UnitTesting; |
|
|
|
using System; |
|
|
|
using System.Collections.Generic; |
|
|
|
using System.Text; |
|
|
|
using Tensorflow; |
|
|
|
|
|
|
|
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)); |
|
|
|
} |
|
|
|
} |
|
|
|
} |