Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/data/experimental/ops/
Ddata_service_ops.py248 num_consumers=None, argument
404 num_consumers, max_outstanding_requests, argument
463 num_consumers=None, argument
559 num_consumers=None, argument
931 num_consumers=None, argument
1090 num_consumers=None, argument
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/
Dtest_base.py245 num_consumers=None, argument
291 num_consumers, argument
347 def checkCoordinatedReadGroups(self, results, num_consumers): argument
Dcoordinated_read_test.py37 def testBasic(self, num_workers, num_consumers): argument
/external/tensorflow/tensorflow/core/data/service/
Dtask_remover.cc29 TaskRemover::TaskRemover(int64_t num_consumers) in TaskRemover()
Dexport.proto17 int64 num_consumers = 6; field
Dcommon.proto42 int64 num_consumers = 7; field
Djournal.proto56 int64 num_consumers = 6; field
Dtask_runner_test.cc504 int64_t num_consumers = std::get<1>(GetParam()); in TEST_P() local
549 int64_t num_consumers = 5; in TEST() local
Dtask_runner.cc206 std::unique_ptr<TaskIterator> iterator, int64_t num_consumers, in RoundRobinTaskRunner()
Ddispatcher_client.cc142 std::optional<int64_t> num_consumers, bool use_cross_trainer_cache, in GetOrCreateJob()
Ddispatcher.proto126 int64 num_consumers = 4; field
Ddispatcher_state.h170 const std::optional<int64_t> num_consumers; member
Ddispatcher_state.cc124 std::optional<int64_t> num_consumers; in CreateJob() local
Ddispatcher_state_test.cc387 int64_t num_consumers = 8; in TEST() local
/external/tensorflow/tensorflow/core/data/service/client/
Dcommon.h39 std::optional<int64_t> num_consumers; member
/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/core/kernels/data/experimental/
Ddata_service_dataset_op.cc186 std::optional<int64_t> num_consumers, in Dataset()
338 Node* num_consumers; in AsGraphDefInternal() local
1388 std::optional<int64_t> num_consumers; in MakeDataset() local
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc568 int num_consumers = TF_OperationOutputNumConsumers(oper_out); in TF_OperationOutputConsumers_wrapper() local
/external/XNNPACK/src/xnnpack/
Dsubgraph.h98 uint32_t num_consumers; member
/external/bcc/libbpf-tools/loongarch/
Dvmlinux_602.h79376 int num_consumers; member
Dvmlinux.h79376 int num_consumers; member
/external/bcc/libbpf-tools/x86/
Dvmlinux.h91648 int num_consumers; member
Dvmlinux_518.h91648 int num_consumers; member
/external/bcc/libbpf-tools/riscv/
Dvmlinux_602.h81350 int num_consumers; member
/external/bcc/libbpf-tools/arm64/
Dvmlinux_608.h44349 int num_consumers; member

12