| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | data_service_ops.py | 64 consumer_index=None, argument
103 if consumer_index is None != num_consumers is None:
106 "consumer_index: ", consumer_index, ", num_consumers: ",
129 -1 if consumer_index is None else consumer_index,
168 consumer_index=self._consumer_index,
188 data_transfer_protocol, job_name, consumer_index, num_consumers, argument
198 consumer_index=consumer_index,
243 consumer_index=None, argument
312 consumer_index=consumer_index,
331 consumer_index=None, argument
[all …]
|
| /external/tensorflow/tensorflow/core/data/service/ |
| D | task_runner.cc | 134 if (req.consumer_index() < 0 || req.round_index() < 0) { in ValidateRequest()
139 if (req.consumer_index() >= num_consumers_) { in ValidateRequest()
141 "Requesting data for consumer index ", req.consumer_index(), in ValidateRequest()
180 round[req.consumer_index()] = &req; in PrepareRound()
182 requests_[req.round_index()].erase(req.consumer_index()); in PrepareRound()
214 VLOG(2) << "Received request from consumer index " << req.consumer_index() in GetNext()
221 << " for consumer " << req.consumer_index(); in GetNext()
225 for (auto& component : buffer_[req.consumer_index()]) { in GetNext()
233 VLOG(2) << "Returning to consumer " << req.consumer_index() << " for round " in GetNext()
|
| D | task_runner_test.cc | 54 Status RunConsumer(int64 consumer_index, int64 start_index, int64 end_index, in RunConsumer() argument
59 request.set_consumer_index(consumer_index); in RunConsumer()
|
| D | worker.proto | 19 int64 consumer_index = 2; field
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.data.experimental.service.pbtxt | 13 …argspec: "args=[\'processing_mode\', \'service\', \'job_name\', \'consumer_index\', \'num_consumer…
17 …de\', \'service\', \'dataset_id\', \'element_spec\', \'job_name\', \'consumer_index\', \'num_consu…
|
| D | tensorflow.raw_ops.pbtxt | 1013 …d\', \'processing_mode\', \'address\', \'protocol\', \'job_name\', \'consumer_index\', \'num_consu…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.data.experimental.service.pbtxt | 21 …argspec: "args=[\'processing_mode\', \'service\', \'job_name\', \'consumer_index\', \'num_consumer…
25 …de\', \'service\', \'dataset_id\', \'element_spec\', \'job_name\', \'consumer_index\', \'num_consu…
|
| D | tensorflow.raw_ops.pbtxt | 1013 …d\', \'processing_mode\', \'address\', \'protocol\', \'job_name\', \'consumer_index\', \'num_consu…
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | data_service_test_base.py | 212 consumer_index=None, argument
221 consumer_index=consumer_index,
|
| D | data_service_ops_test.py | 297 for consumer_index in range(num_consumers):
303 consumer_index=consumer_index,
360 consumer_index = host_index * replicas_per_consumer_host + i
366 consumer_index=consumer_index,
413 ds, cluster, job_name="test", consumer_index=0, num_consumers=1)
426 ds, cluster, job_name="test", consumer_index=0, num_consumers=1)
|
| D | data_service_ops_ft_test.py | 164 for consumer_index in range(num_consumers):
170 consumer_index=consumer_index,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | DataServiceDatasetV2.pbtxt | 24 name: "consumer_index"
87 name: "consumer_index"
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | data_service_dataset_op.cc | 90 const std::string& job_name, absl::optional<int64> consumer_index, in Dataset() argument
104 consumer_index_(consumer_index), in Dataset()
182 Node* consumer_index; in AsGraphDefInternal() local
184 b->AddScalar(consumer_index_.value_or(-1), &consumer_index)); in AsGraphDefInternal()
185 inputs.push_back(consumer_index); in AsGraphDefInternal()
907 absl::optional<int64> consumer_index; in MakeDataset() local
914 consumer_index = consumer_index_int; in MakeDataset()
966 consumer_index, num_consumers, max_outstanding_requests, in MakeDataset()
|
| /external/kernel-headers/original/uapi/rdma/ |
| D | rdma_user_rxe.h | 200 __u32 consumer_index; member
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 48930 …f.Output, address tf.Output, protocol tf.Output, job_name tf.Output, consumer_index tf.Output, num…
48941 …dataset_id, processing_mode, address, protocol, job_name, consumer_index, num_consumers, max_outst…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 11253 name: "consumer_index"
|