data_service_client.h
imports tensorflow/core/platform/env.himports tensorflow/core/platform/mutex.h
SHA-2568defebccad788c3c7bff925731f5152a2e80aba3bd2877bb3fb2341f4a29f775
MaleculeMd(In₅)
Evidence
1/* Copyright 2022 The TensorFlow Authors. All Rights Reserved.
2
3Licensed under the Apache License, Version 2.0 (the "License");
4you may not use this file except in compliance with the License.
5You may obtain a copy of the License at
6
7 http://www.apache.org/licenses/LICENSE-2.0
8
9Unless required by applicable law or agreed to in writing, software
10distributed under the License is …
11:42… IND, either express or implied.
12See the License for the specific language governing permissions and
13limitations under the License.
14================================================================ …
16:20… CORE_DATA_SERVICE_CLIENT_DATA_SERVICE_CLIENT_H_
17
18#include <cstdint>
19#include <functional>
20#include <memory>
21#include <optional>
22#include <queue>
23#include <string>
24#include <utility>
25#include <vector>
26
27#include "absl/container/flat_hash_set.h"
28#include "absl/status/status.h"
29#include "absl/status/statusor.h"
30#include "tensorflow/core/data/service/client/common.h"
31#include "tensorflow/core/data/service/common.h"
32#include "tensorflow/core/data/service/common.pb.h"
33#include "tensorflow/core/data/service/dispatcher.pb.h"
34#include "tensorflow/core/data/service/dispatcher_client.h"
35#include "tensorflow/core/data/service/worker_client.h"
36#include "tensorflow/core/framework/dataset.h"
37#include "tensorflow/core/framework/tensor.h"
38#include "tensorflow/core/platform/env.h"
39#include "tensorflow/core/platform/mutex.h"
40#include "tensorflow/core/platform/status.h"
41#include "tensorflow/core/platform/statusor.h"
42#include "tensorflow/core/platform/thread_annotations.h"
43
44namespace tensorflow {
⋯4 lines
55:56… peline can take to
56 // produce an element such that the downstream processor wait time is 0.
57 // Returns 0 if there are not sufficient recorded it …
152:53… task.
153 // TODO(aaudibert): Instead of polling, have dispatcher send updates when
154 // the list of tasks changes.
155 void TaskThreadManager();
156 void …