Function TEST

tensorflow/core/framework/unique_tensor_references_test.cc:23–44

namespace tensorflow {

TEST(UniquifyTensors, TestUniqueVector) {
  UniqueTensorReferences refs;
  Tensor a(DT_FLOAT, TensorShape({2, 2}));
  Tensor b(DT_FLOAT, TensorShape({2, 2}));

  EXPECT_FALSE(a.SharesBufferWith(b));

  refs.Add(a);
  refs.Add(b);
  TensorReferenceVector tensors;
  refs.FreezeAndReturnReferences(&tensors);
  EXPECT_EQ(2, tensors.size());
  if (tensors[0].SharesBufferWith(a)) {
    EXPECT_TRUE(tensors[1].SharesBufferWith(b));
  } else {
    EXPECT_TRUE(tensors[1].SharesBufferWith(a));
    EXPECT_TRUE(tensors[0].SharesBufferWith(b));
  }
  for (auto& t : tensors) {
    t.Unref();
  }
}

TEST(UniquifyTensors, TestNonUniqueVector) {
  UniqueTensorReferences refs;