/external/tensorflow/tensorflow/python/data/experimental/ops/ |
D | data_service_ops.py | 65 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/ |
D | common.proto | 26 int64 num_consumers = 7; field
|
D | task_runner_test.cc | 104 int64 num_consumers = std::get<1>(GetParam()); in TEST_P() local 153 int64 num_consumers = 5; in TEST() local
|
D | journal.proto | 49 int64 num_consumers = 7; field
|
D | task_runner.cc | 125 std::unique_ptr<TaskIterator> iterator, int64 num_consumers) in RoundRobinTaskRunner()
|
D | dispatcher.proto | 87 int64 num_consumers = 7; field
|
D | dispatcher_state.h | 144 absl::optional<int64> num_consumers; member
|
D | dispatcher_state.cc | 96 absl::optional<int64> num_consumers; in CreateJob() local
|
D | dispatcher_impl.cc | 451 absl::optional<int64> num_consumers; in GetOrCreateJob() local 509 absl::optional<int64> num_consumers, std::shared_ptr<const Job>& job) in CreateJob()
|
D | data_service.cc | 158 const absl::optional<JobKey>& job_key, absl::optional<int64> num_consumers, in GetOrCreateJob()
|
D | dispatcher_state_test.cc | 245 int64 num_consumers = 8; in TEST() local
|
/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
D | data_service_dataset_op.cc | 91 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/ |
D | data_service_test_base.py | 213 num_consumers=None, argument
|
D | data_service_ops_test.py | 290 def testRoundRobin(self, num_workers, num_consumers): argument
|
/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
D | gpu_allocator_retry_test.cc | 123 void LaunchConsumerThreads(int num_consumers, int cap_needed) { in LaunchConsumerThreads()
|
/external/tensorflow/tensorflow/python/client/ |
D | tf_session_helper.cc | 567 int num_consumers = TF_OperationOutputNumConsumers(oper_out); in TF_OperationOutputConsumers_wrapper() local
|
/external/XNNPACK/src/xnnpack/ |
D | subgraph.h | 67 uint32_t num_consumers; member
|