Searched refs:serialization_ctx (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | concatenate_dataset_op_test.cc | 267 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_F() local
268 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_F()
271 TF_ASSERT_OK(dataset_->Save(serialization_ctx.get(), &writer)); in TEST_F()
311 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_P() local
312 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_P()
321 TF_EXPECT_OK(iterator_->Save(serialization_ctx.get(), &writer)); in TEST_P()
|
| D | zip_dataset_op_test.cc | 251 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_F() local
252 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_F()
255 TF_ASSERT_OK(dataset_->Save(serialization_ctx.get(), &writer)); in TEST_F()
311 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_P() local
312 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_P()
320 TF_EXPECT_OK(iterator_->Save(serialization_ctx.get(), &writer)); in TEST_P()
|
| D | dataset_ops.cc | 38 SerializationContext serialization_ctx(params); in Compute() local
40 ctx, db.AddInputDataset(&serialization_ctx, dataset, &input_node)); in Compute()
|
| D | sparse_tensor_slice_dataset_op_test.cc | 319 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_F() local
320 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_F()
323 TF_ASSERT_OK(dataset->Save(serialization_ctx.get(), &writer)); in TEST_F()
468 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_F() local
469 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_F()
500 TF_ASSERT_OK(iterator->Save(serialization_ctx.get(), &writer)); in TEST_F()
|
| D | take_dataset_op_test.cc | 356 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_F() local
357 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_F()
360 TF_ASSERT_OK(take_dataset->Save(serialization_ctx.get(), &writer)); in TEST_F()
516 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_P() local
517 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_P()
526 TF_EXPECT_OK(iterator->Save(serialization_ctx.get(), &writer)); in TEST_P()
|
| D | repeat_dataset_op_test.cc | 349 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_F() local
350 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_F()
353 TF_ASSERT_OK(repeat_dataset->Save(serialization_ctx.get(), &writer)); in TEST_F()
508 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_P() local
509 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_P()
518 TF_EXPECT_OK(iterator->Save(serialization_ctx.get(), &writer)); in TEST_P()
|
| D | graph_rewrite_dataset.cc | 41 SerializationContext serialization_ctx(params); in Optimize() local
43 db.AddInputDataset(&serialization_ctx, input_, &input_node)); in Optimize()
|
| D | tensor_dataset_op_test.cc | 483 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_P() local
484 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_P()
492 TF_EXPECT_OK(iterator->Save(serialization_ctx.get(), &writer)); in TEST_P()
|
| D | iterator_ops.cc | 342 SerializationContext serialization_ctx(params); in InitializeFromIterator() local
346 TF_RETURN_IF_ERROR(iterator_resource->Save(&serialization_ctx, &writer)); in InitializeFromIterator()
|