Home
last modified time | relevance | path

Searched refs:serialization_ctx (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dserialization_utils.cc66 SerializationContext serialization_ctx(params); in AsGraphDefMinimal() local
68 AsGraphDef(ctx, input, std::move(serialization_ctx), result)); in AsGraphDefMinimal()
80 SerializationContext&& serialization_ctx, in AsGraphDef() argument
82 if (serialization_ctx.external_state_policy() == in AsGraphDef()
86 if (serialization_ctx.external_state_policy() == in AsGraphDef()
102 db.AddInputDataset(&serialization_ctx, dataset, &output_node)); in AsGraphDef()
Dtensor_dataset_op_test.cc230 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_P() local
231 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_P()
239 TF_EXPECT_OK(iterator_->Save(serialization_ctx.get(), &writer)); in TEST_P()
Drepeat_dataset_op_test.cc312 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_P() local
313 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_P()
322 TF_EXPECT_OK(iterator_->Save(serialization_ctx.get(), &writer)); in TEST_P()
Dcache_dataset_ops_test.cc329 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_P() local
330 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_P()
335 TF_EXPECT_OK(iterator_->Save(serialization_ctx.get(), &writer)); in TEST_P()
Dserialization_utils.h27 SerializationContext&& serialization_ctx,
Dsparse_tensor_slice_dataset_op_test.cc371 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_P() local
372 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_P()
402 TF_ASSERT_OK(iterator_->Save(serialization_ctx.get(), &writer)); in TEST_P()
Dshuffle_dataset_op_test.cc518 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_P() local
519 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_P()
527 TF_EXPECT_OK(iterator_->Save(serialization_ctx.get(), &writer)); in TEST_P()
Dwindow_dataset_op_test.cc469 std::unique_ptr<SerializationContext> serialization_ctx; in TEST_P() local
470 TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx)); in TEST_P()
477 TF_EXPECT_OK(iterator_->Save(serialization_ctx.get(), &writer)); in TEST_P()
Diterator_ops.cc370 Status InitializeFromIterator(SerializationContext* serialization_ctx, in InitializeFromIterator() argument
373 TF_RETURN_IF_ERROR(iterator_resource->Save(serialization_ctx, &writer)); in InitializeFromIterator()
1144 SerializationContext serialization_ctx(params); in Compute() local
1145 OP_REQUIRES_OK(ctx, serializer.InitializeFromIterator(&serialization_ctx, in Compute()
Ddataset_test_base.cc731 std::unique_ptr<SerializationContext> serialization_ctx; in CheckIteratorSaveAndRestore() local
732 TF_RETURN_IF_ERROR(CreateSerializationContext(&serialization_ctx)); in CheckIteratorSaveAndRestore()
738 TF_EXPECT_OK(iterator->Save(serialization_ctx.get(), &writer)); in CheckIteratorSaveAndRestore()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddata_service_ops.cc56 SerializationContext serialization_ctx(params); in Compute() local
59 ctx, AsGraphDef(ctx, dataset, std::move(serialization_ctx), &graph_def)); in Compute()
/external/tensorflow/tensorflow/core/framework/
Ddataset.h908 SerializationContext&& serialization_ctx,