Home
last modified time | relevance | path

Searched defs:num_consumers (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/ops/
Ddata_service_ops.py65 num_consumers=None, argument
188 data_transfer_protocol, job_name, consumer_index, num_consumers, argument
244 num_consumers=None, argument
332 num_consumers=None, argument
400 num_consumers=None, argument
689 num_consumers=None, argument
/external/tensorflow/tensorflow/core/data/service/
Dcommon.proto26 int64 num_consumers = 7; field
Dtask_runner_test.cc104 int64 num_consumers = std::get<1>(GetParam()); in TEST_P() local
153 int64 num_consumers = 5; in TEST() local
Djournal.proto49 int64 num_consumers = 7; field
Dtask_runner.cc125 std::unique_ptr<TaskIterator> iterator, int64 num_consumers) in RoundRobinTaskRunner()
Ddispatcher.proto87 int64 num_consumers = 7; field
Ddispatcher_state.h144 absl::optional<int64> num_consumers; member
Ddispatcher_state.cc96 absl::optional<int64> num_consumers; in CreateJob() local
Ddispatcher_impl.cc451 absl::optional<int64> num_consumers; in GetOrCreateJob() local
509 absl::optional<int64> num_consumers, std::shared_ptr<const Job>& job) in CreateJob()
Ddata_service.cc158 const absl::optional<JobKey>& job_key, absl::optional<int64> num_consumers, in GetOrCreateJob()
Ddispatcher_state_test.cc245 int64 num_consumers = 8; in TEST() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddata_service_dataset_op.cc91 absl::optional<int64> num_consumers, int64 max_outstanding_requests, in Dataset()
187 Node* num_consumers; in AsGraphDefInternal() local
908 absl::optional<int64> num_consumers; in MakeDataset() local
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Ddata_service_test_base.py213 num_consumers=None, argument
Ddata_service_ops_test.py290 def testRoundRobin(self, num_workers, num_consumers): argument
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_allocator_retry_test.cc123 void LaunchConsumerThreads(int num_consumers, int cap_needed) { in LaunchConsumerThreads()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc567 int num_consumers = TF_OperationOutputNumConsumers(oper_out); in TF_OperationOutputConsumers_wrapper() local
/external/XNNPACK/src/xnnpack/
Dsubgraph.h67 uint32_t num_consumers; member