Searched refs:serialization_ctx (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | serialization_utils.cc | 66 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()
|
| D | tensor_dataset_op_test.cc | 230 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()
|
| D | repeat_dataset_op_test.cc | 312 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()
|
| D | cache_dataset_ops_test.cc | 329 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()
|
| D | serialization_utils.h | 27 SerializationContext&& serialization_ctx,
|
| D | sparse_tensor_slice_dataset_op_test.cc | 371 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()
|
| D | shuffle_dataset_op_test.cc | 518 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()
|
| D | window_dataset_op_test.cc | 469 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()
|
| D | iterator_ops.cc | 370 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()
|
| D | dataset_test_base.cc | 731 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/ |
| D | data_service_ops.cc | 56 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/ |
| D | dataset.h | 908 SerializationContext&& serialization_ctx,
|