| 90 | }; |
| 91 | |
| 92 | TEST_F(SummaryFileWriterTest, WriteTensor) { |
| 93 | TF_CHECK_OK(SummaryTestHelper("tensor_test", |
| 94 | [](SummaryWriterInterface* writer) { |
| 95 | Tensor one(DT_FLOAT, TensorShape({})); |
| 96 | one.scalar<float>()() = 1.0; |
| 97 | TF_RETURN_IF_ERROR(writer->WriteTensor( |
| 98 | 2, one, "name", |
| 99 | SummaryMetadata().SerializeAsString())); |
| 100 | TF_RETURN_IF_ERROR(writer->Flush()); |
| 101 | return Status::OK(); |
| 102 | }, |
| 103 | [](const Event& e) { |
| 104 | EXPECT_EQ(e.step(), 2); |
| 105 | CHECK_EQ(e.summary().value_size(), 1); |
| 106 | EXPECT_EQ(e.summary().value(0).tag(), "name"); |
| 107 | })); |
| 108 | TF_CHECK_OK(SummaryTestHelper( |
| 109 | "string_tensor_test", |
| 110 | [](SummaryWriterInterface* writer) { |
| 111 | Tensor hello(DT_STRING, TensorShape({})); |
| 112 | hello.scalar<tstring>()() = "hello"; |
| 113 | TF_RETURN_IF_ERROR(writer->WriteTensor( |
| 114 | 2, hello, "name", SummaryMetadata().SerializeAsString())); |
| 115 | TF_RETURN_IF_ERROR(writer->Flush()); |
| 116 | return Status::OK(); |
| 117 | }, |
| 118 | [](const Event& e) { |
| 119 | EXPECT_EQ(e.step(), 2); |
| 120 | CHECK_EQ(e.summary().value_size(), 1); |
| 121 | EXPECT_EQ(e.summary().value(0).tag(), "name"); |
| 122 | EXPECT_EQ(e.summary().value(0).tensor().dtype(), DT_STRING); |
| 123 | EXPECT_EQ(e.summary().value(0).tensor().string_val()[0], "hello"); |
| 124 | })); |
| 125 | } |
| 126 | |
| 127 | TEST_F(SummaryFileWriterTest, WriteScalar) { |
| 128 | TF_CHECK_OK(SummaryTestHelper( |
nothing calls this directly
no test coverage detected