| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/ops/ |
| H A D | data_service_ops.py | 248 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/ |
| H A D | test_base.py | 245 num_consumers=None, argument
291 num_consumers, argument
347 def checkCoordinatedReadGroups(self, results, num_consumers): argument
|
| H A D | coordinated_read_test.py | 37 def testBasic(self, num_workers, num_consumers): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | data_service_dataset_op.cc | 186 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
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/service/ |
| H A D | task_remover.cc | 29 TaskRemover::TaskRemover(int64_t num_consumers) in TaskRemover()
|
| H A D | task_runner_test.cc | 504 int64_t num_consumers = std::get<1>(GetParam()); in TEST_P() local
549 int64_t num_consumers = 5; in TEST() local
|
| H A D | export.proto | 17 int64 num_consumers = 6; field
|
| H A D | common.proto | 42 int64 num_consumers = 7; field
|
| H A D | journal.proto | 56 int64 num_consumers = 6; field
|
| H A D | task_runner.cc | 206 std::unique_ptr<TaskIterator> iterator, int64_t num_consumers, in RoundRobinTaskRunner()
|
| H A D | dispatcher_client.cc | 142 std::optional<int64_t> num_consumers, bool use_cross_trainer_cache, in GetOrCreateJob()
|
| H A D | dispatcher.proto | 126 int64 num_consumers = 4; field
|
| H A D | dispatcher_state.h | 170 const std::optional<int64_t> num_consumers; member
|
| H A D | dispatcher_state.cc | 124 std::optional<int64_t> num_consumers; in CreateJob() local
|
| H A D | dispatcher_state_test.cc | 387 int64_t num_consumers = 8; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/data/service/client/ |
| H A D | common.h | 39 std::optional<int64_t> num_consumers; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_allocator_retry_test.cc | 123 void LaunchConsumerThreads(int num_consumers, int cap_needed) { in LaunchConsumerThreads()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/client/ |
| H A D | tf_session_helper.cc | 568 int num_consumers = TF_OperationOutputNumConsumers(oper_out); in TF_OperationOutputConsumers_wrapper() local
|
| /aosp_15_r20/external/XNNPACK/src/xnnpack/ |
| H A D | subgraph.h | 98 uint32_t num_consumers; member
|
| /aosp_15_r20/external/bcc/libbpf-tools/loongarch/ |
| H A D | vmlinux.h | 79376 int num_consumers; member
|
| H A D | vmlinux_602.h | 79376 int num_consumers; member
|
| /aosp_15_r20/external/bcc/libbpf-tools/x86/ |
| H A D | vmlinux.h | 91648 int num_consumers; member
|
| H A D | vmlinux_518.h | 91648 int num_consumers; member
|
| /aosp_15_r20/system/bpf/include/vmlinux/android15_6.6/arm64/ |
| H A D | vmlinux.h | 73822 int num_consumers; member
|
| /aosp_15_r20/external/bcc/libbpf-tools/arm64/ |
| H A D | vmlinux_608.h | 44349 int num_consumers; member
|