diff options
| author |  Mingsheng Hong <hongm@google.com> | 2018-03-26 11:50:19 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-03-26 11:53:04 -0700 |
| commit | f9cfb9e917c8937152b248c300b095798d79501a (patch) | |
| tree | c269dd414b3dded2d8b6a2dc89010c7540926369 /tensorflow/c/c_api_experimental_test.cc | |
| parent | d2604f8dcb8a63ca063f712c24ce5aa63403b0aa (diff) | |
Extended experimental C API with MNIST dataset/iterators support.
PiperOrigin-RevId: 190500020
Diffstat (limited to 'tensorflow/c/c_api_experimental_test.cc')
| -rw-r--r-- | tensorflow/c/c_api_experimental_test.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/c/c_api_experimental_test.cc b/tensorflow/c/c_api_experimental_test.cc index 49d64d18bf..30fcfd401d 100644 --- a/tensorflow/c/c_api_experimental_test.cc +++ b/tensorflow/c/c_api_experimental_test.cc @@ -68,8 +68,8 @@ TEST(CAPI_EXPERIMENTAL, ImagenetIteratorGetNext) { tensorflow::testing::TensorFlowSrcRoot(), "c/testdata/tf_record"); VLOG(1) << "data file path is " << file_path; const int batch_size = 64; - TF_Operation* get_next = TF_MakeImagenetIteratorGetNextWithDatasets( - graph, file_path.c_str(), batch_size, s); + TF_Operation* get_next = TF_MakeFileBasedIteratorGetNextWithDatasets( + graph, file_path.c_str(), batch_size, /*is_mnist*/ false, s); ASSERT_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); CSession csession(graph, s); |