| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | parameter_server_strategy_test.py | 72 num_gpus=None, argument
111 def _get_test_objects(self, task_type, task_id, num_gpus): argument
119 def _test_device_assignment_distributed(self, task_type, task_id, num_gpus): argument
219 self, task_type, task_id, num_gpus): argument
270 num_gpus=0): argument
368 def _test_simple_increment(self, task_type, task_id, num_gpus): argument
432 def _test_minimize_loss_graph(self, task_type, task_id, num_gpus): argument
521 num_gpus, argument
603 def testDeviceAssignmentDistributed(self, num_gpus): argument
608 def testDeviceAssignmentDistributedEnablePartitioner(self, num_gpus): argument
[all …]
|
| H A D | collective_all_reduce_strategy_test.py | 74 num_gpus=None, argument
113 num_gpus=0, argument
138 def _test_minimize_loss_graph(self, task_type, task_id, num_gpus): argument
211 def _test_variable_initialization(self, task_type, task_id, num_gpus): argument
241 num_gpus, argument
|
| H A D | multi_worker_test_base.py | 511 def _run_client(self, client_fn, task_type, task_id, num_gpus, eager_mode, argument
525 def _run_between_graph_clients(self, client_fn, cluster_spec, num_gpus, *args, argument
|
| H A D | device_util.py | 157 def local_devices_from_num_gpus(num_gpus): argument
|
| H A D | central_storage_strategy.py | 70 def _from_num_gpus(cls, num_gpus): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/v1/ |
| H A D | all_reduce_test.py | 85 def _buildInput(self, num_workers, num_gpus): argument
140 def _buildRing(self, num_workers, num_gpus, subdiv): argument
145 def _testAllReduce(self, num_workers, num_gpus, shape, build_f): argument
161 def _testRingAllReduce(self, num_workers, num_gpus, shape, subdiv): argument
185 def _buildShuffle(self, num_workers, num_gpus, num_shards): argument
192 def _testShuffleAllReduce(self, num_workers, num_gpus, shape, num_shards): argument
212 def _buildRecursiveHD(self, num_workers, num_gpus): argument
218 def _testRecursiveHDAllReduce(self, num_workers, num_gpus, shape): argument
|
| H A D | cross_device_ops_test.py | 457 num_gpus=0, argument
530 num_gpus, argument
669 num_gpus, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | generic_layout_optimizer.cc | 51 int num_gpus = 0; in GetNumGPUs() local
108 const TransposeContext& context, int num_gpus, int num_voltas) { in GetSrcAndDstDataFormats()
426 const int num_gpus = num_gpus_and_num_volta.first; in Optimize() local
|
| /aosp_15_r20/external/mesa3d/src/amd/addrlib/src/chip/r800/ |
| H A D | si_gb_reg.h | 46 unsigned int num_gpus : 3; member
64 unsigned int num_gpus : 3; member
91 unsigned int num_gpus : 3; member
107 unsigned int num_gpus : 3; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_testlib_server.cc | 39 int num_cpus, int num_gpus, int task_index, in FillServerDef()
83 int num_gpus = 0; in main() local
|
| H A D | grpc_testlib.cc | 52 int num_gpus = 0; in MakeTestCluster() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| H A D | cupti_collector.h | 205 uint32 num_gpus; member
215 explicit AnnotationMap(uint64 max_size, uint32 num_gpus) in AnnotationMap()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | compiled_collective_ops_gpu_test.py | 44 def _setup_context(self, num_gpus=2): argument
|
| H A D | collective_ops_gpu_test.py | 46 def _setup_context(self, num_gpus=2): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A D | devices.cc | 46 int num_gpus = gpu_manager->VisibleDeviceCount(); in GetNumAvailableGPUs() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/api/src/ |
| H A D | cuda.cpp | 56 int64_t num_gpus = cuda::device_count(); in synchronize() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDAContext.cpp | 13 DeviceIndex num_gpus = -1; variable
|
| H A D | CUDAGeneratorImpl.cpp | 21 static int64_t num_gpus; variable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/xpu/ |
| H A D | XPUContext.cpp | 18 DeviceIndex num_gpus = -1; variable
|
| H A D | XPUGeneratorImpl.cpp | 17 DeviceIndex num_gpus = -1; variable
|
| /aosp_15_r20/hardware/google/gfxstream/host/gl/gl-host-common/include/host-common/opengl/ |
| D | emugl_config.h | 76 int num_gpus; member
|
| /aosp_15_r20/external/pytorch/c10/xpu/ |
| H A D | XPUStream.cpp | 16 DeviceIndex num_gpus = -1; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | rpcbench_test.cc | 57 int num_gpus = 0; in MakeGRPCCluster() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/debug/ |
| H A D | grpc_session_debug_test.cc | 45 SessionOptions Devices(int num_cpus, int num_gpus) { in Devices()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/ |
| H A D | single_machine.cc | 41 SingleMachine::SingleMachine(int timeout_s, int num_cpu_cores, int num_gpus) in SingleMachine()
|