Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/data/service/
Ddataset_store_test.cc68 DatasetDef dataset_def = DatasetDefWithVersion(1); in TEST_P() local
69 TF_ASSERT_OK(store->Put(key, dataset_def)); in TEST_P()
72 EXPECT_EQ(result->graph().version(), dataset_def.graph().version()); in TEST_P()
81 DatasetDef dataset_def = DatasetDefWithVersion(i); in TEST_P() local
82 TF_ASSERT_OK(store->Put(key, dataset_def)); in TEST_P()
95 DatasetDef dataset_def = DatasetDefWithVersion(version); in TEST_P() local
97 TF_ASSERT_OK(store->Put(key, dataset_def)); in TEST_P()
98 Status s = store->Put(key, dataset_def); in TEST_P()
Dutils.cc28 Status WriteDatasetDef(const std::string& path, const DatasetDef& dataset_def) { in WriteDatasetDef() argument
32 TF_RETURN_IF_ERROR(writer.WriteRecord(dataset_def.SerializeAsString())); in WriteDatasetDef()
36 Status ReadDatasetDef(const std::string& path, DatasetDef& dataset_def) { in ReadDatasetDef() argument
47 if (!dataset_def.ParseFromString(record)) { in ReadDatasetDef()
Ddataset_store.cc48 const std::string& key, std::shared_ptr<const DatasetDef>& dataset_def) { in Get() argument
53 dataset_def = std::make_shared<const DatasetDef>(def); in Get()
71 std::shared_ptr<const DatasetDef>& dataset_def) { in Get() argument
76 dataset_def = stored_dataset; in Get()
Ddataset_store.h38 std::shared_ptr<const DatasetDef>& dataset_def) = 0;
51 std::shared_ptr<const DatasetDef>& dataset_def) override;
67 std::shared_ptr<const DatasetDef>& dataset_def) override;
Ddispatcher_impl.cc244 std::shared_ptr<const DatasetDef> dataset_def; in WorkerHeartbeat() local
245 TF_RETURN_IF_ERROR(dataset_store_->Get(dataset_key, dataset_def)); in WorkerHeartbeat()
246 *task_def->mutable_dataset_def() = *dataset_def; in WorkerHeartbeat()
302 std::shared_ptr<const DatasetDef> dataset_def; in GetDatasetDef() local
303 TF_RETURN_IF_ERROR(GetDatasetDef(*dataset, dataset_def)); in GetDatasetDef()
304 *response->mutable_dataset_def() = *dataset_def; in GetDatasetDef()
354 std::shared_ptr<const DatasetDef> dataset_def; in MakeSplitProvider() local
355 TF_RETURN_IF_ERROR(GetDatasetDef(*dataset, dataset_def)); in MakeSplitProvider()
359 params, dataset_def->graph(), &standalone_dataset)); in MakeSplitProvider()
375 DatasetDef dataset_def = request->dataset(); in GetOrRegisterDataset() local
[all …]
Dutils.h29 Status WriteDatasetDef(const std::string& path, const DatasetDef& dataset_def);
33 Status ReadDatasetDef(const std::string& path, DatasetDef& dataset_def);
Ddispatcher_impl.h235 std::shared_ptr<const DatasetDef>& dataset_def)
240 std::shared_ptr<const DatasetDef>& dataset_def)
Ddata_service.cc105 DatasetDef& dataset_def) { in GetDatasetDef() argument
115 dataset_def = resp.dataset_def(); in GetDatasetDef()
Dcommon.proto16 DatasetDef dataset_def = 1; field
Ddata_service.h102 Status GetDatasetDef(int64 dataset_id, DatasetDef& dataset_def);
Ddispatcher.proto38 DatasetDef dataset_def = 1; field
Dworker_impl.cc135 params, task.task_def.dataset_def().graph(), &dataset)); in EnsureTaskInitialized()