You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

ConsumersTest.cs 848 B

1234567891011121314151617181920212223242526272829303132
  1. using Microsoft.VisualStudio.TestTools.UnitTesting;
  2. using Tensorflow;
  3. using static Tensorflow.Binding;
  4. namespace TensorFlowNET.UnitTest
  5. {
  6. [TestClass]
  7. public class ConsumersTest : CApiTest
  8. {
  9. [TestMethod]
  10. public void Constant()
  11. {
  12. var X = tf.placeholder(tf.float64);
  13. var W = tf.constant(1.0D);
  14. var mul = tf.multiply(X, W);
  15. EXPECT_EQ(1, X.op.OutputNumConsumers(0));
  16. EXPECT_EQ(1, W.op.OutputNumConsumers(0));
  17. }
  18. [TestMethod]
  19. public void Variable()
  20. {
  21. var X = tf.placeholder(tf.float64);
  22. var W = tf.Variable(1.0D, name: "var");
  23. var mul = tf.multiply(X, W);
  24. EXPECT_EQ(1, X.op.OutputNumConsumers(0));
  25. //EXPECT_EQ(1, W.op.OutputNumConsumers(0));
  26. }
  27. }
  28. }