forked from SciSharp/TensorFlow.NET
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathConsumersTest.cs
More file actions
32 lines (28 loc) · 848 Bytes
/
ConsumersTest.cs
File metadata and controls
32 lines (28 loc) · 848 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
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));
}
}
}