Searched refs:num_worker_threads (Results 1 – 22 of 22) sorted by relevance
| /external/pytorch/torch/distributed/rpc/_testing/ |
| D | faulty_agent_backend_registry.py | 11 num_worker_threads, argument
20 num_worker_threads=num_worker_threads,
|
| /external/pytorch/torch/distributed/rpc/ |
| D | options.py | 83 num_worker_threads: int = rpc_contants.DEFAULT_NUM_WORKER_THREADS,
98 num_worker_threads,
|
| D | backend_registry.py | 138 num_worker_threads=rpc_constants.DEFAULT_NUM_WORKER_THREADS, argument
148 num_worker_threads=num_worker_threads,
|
| /external/pytorch/torch/csrc/distributed/rpc/testing/ |
| D | faulty_tensorpipe_agent.h | 15 int num_worker_threads,
22 num_worker_threads, in TensorPipeRpcBackendOptions()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/benchmarks/ |
| D | benchmark.cc | 62 copts.num_worker_threads = host.GetNumWorkerThreads(); in CreateJitExecutable()
86 llvm::formatv("{0}/{1}/{2}", mlir_input.data(), copts.num_worker_threads, in CreateJitExecutable()
|
| /external/pytorch/torch/testing/_internal/distributed/rpc/ |
| D | faulty_rpc_agent_test_fixture.py | 41 num_worker_threads=8,
|
| D | faulty_agent_rpc_test.py | 50 self.assertEqual(self.rpc_backend_options.num_worker_threads, 8)
|
| D | rpc_test.py | 4673 num_worker_threads=NUM_THREADS,
4696 num_worker_threads=self.rpc_backend_options.num_worker_threads,
4722 num_worker_threads=self.rpc_backend_options.num_worker_threads,
5640 num_worker_threads=options.num_worker_threads,
5773 for _ in range(self.rpc_backend_options.num_worker_threads + 5):
5784 for _ in range(self.rpc_backend_options.num_worker_threads + 5):
6270 num_worker_threads=self.rpc_backend_options.num_worker_threads,
|
| D | dist_autograd_test.py | 2635 self.assertEqual(self.rpc_backend_options.num_worker_threads, 8)
|
| /external/pytorch/torch/_C/ |
| D | _distributed_rpc_testing.pyi | 10 num_worker_threads: int,
|
| D | _distributed_rpc.pyi | 71 num_worker_threads: int
76 num_worker_threads: int,
|
| D | __init__.pyi.in | 1652 num_worker_threads: _int
|
| /external/pytorch/torch/csrc/utils/ |
| D | init.cpp | 17 .def_readwrite("num_worker_threads", &BenchmarkConfig::num_worker_threads) in initThroughputBenchmarkBindings()
|
| D | throughput_benchmark.h | 50 int num_worker_threads{1};
|
| D | throughput_benchmark-inl.h | 22 config.num_worker_threads == 1, in benchmark()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorContractionThreadPool.h | 236 const int num_worker_threads = this->m_device.numThreadsInPool();
242 num_worker_threads <= 4 ? 8.0 :
243 num_worker_threads <= 8 ? 4.0 :
244 num_worker_threads <= 16 ? 2.0 :
245 num_worker_threads <= 32 ? 1.0 :
246 num_worker_threads <= 64 ? 0.8 : /* num_worker_threads > 64 */ 0.6;
249 sharding_dim_tasks >= oversharding_factor * num_worker_threads;
463 const int num_worker_threads = device_.numThreadsInPool();
471 Index num_blocks = num_worker_threads * gn_;
485 Index num_blocks = num_worker_threads * gm_;
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/tests/jit/ |
| D | tf_jitrt_benchmark_test.cc | 74 copts.num_worker_threads = 16; in BM_InstantiateExecutable()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/python_binding/ |
| D | tf_jitrt_executor.cc | 84 copts.num_worker_threads = 4; in Compile()
|
| D | tf_cpurt_executor.cc | 74 opts.num_worker_threads = 4; in Compile()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| D | tf_cpurt_kernels.cc | 160 opts.num_worker_threads = workers->NumThreads(); in CompileImpl()
|
| D | tf_jitrt_kernels.cc | 438 copts.num_worker_threads = workers->NumThreads(); in CompileImpl()
|
| /external/pytorch/docs/source/ |
| D | rpc.rst | 182 >>> num_worker_threads=8,
|