| /external/tensorflow/tensorflow/python/distribute/v1/ |
| 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
|
| D | cross_device_ops_test.py | 457 num_gpus=0, argument
530 num_gpus, argument
669 num_gpus, argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| 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 …]
|
| 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
|
| D | device_util.py | 157 def local_devices_from_num_gpus(num_gpus): argument
|
| D | central_storage_strategy.py | 70 def _from_num_gpus(cls, num_gpus): argument
|
| 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
|
| D | mirrored_strategy.py | 183 def all_local_devices(num_gpus=None): argument
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| 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
|
| D | grpc_testlib.cc | 52 int num_gpus = 0; in MakeTestCluster() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | compiled_collective_ops_gpu_test.py | 44 def _setup_context(self, num_gpus=2): argument
|
| D | collective_ops_gpu_test.py | 46 def _setup_context(self, num_gpus=2): argument
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | devices.cc | 46 int num_gpus = gpu_manager->VisibleDeviceCount(); in GetNumAvailableGPUs() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| 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
|
| /external/tensorflow/tensorflow/core/profiler/backends/gpu/ |
| D | cupti_collector.h | 205 uint32 num_gpus; member
215 explicit AnnotationMap(uint64 max_size, uint32 num_gpus) in AnnotationMap()
|
| D | rocm_tracer.h | 183 uint32_t num_gpus; member
|
| D | device_tracer_rocm.cc | 1029 uint32_t num_gpus) { in GetRocmTraceCollectorOptions()
|
| D | rocm_tracer.cc | 1355 static int num_gpus = []() -> int { in NumGpus() local
|
| /external/mesa3d/src/amd/addrlib/src/chip/r800/ |
| D | si_gb_reg.h | 64 unsigned int num_gpus : 3; member
82 unsigned int num_gpus : 3; member
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | rpcbench_test.cc | 57 int num_gpus = 0; in MakeGRPCCluster() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 41 SingleMachine::SingleMachine(int timeout_s, int num_cpu_cores, int num_gpus) in SingleMachine()
|
| /external/tensorflow/tensorflow/core/debug/ |
| D | grpc_session_debug_test.cc | 45 SessionOptions Devices(int num_cpus, int num_gpus) { in Devices()
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | cluster_wrapper.cc | 84 int num_gpus = tensorflow::grappler::GetNumAvailableGPUs(); in PYBIND11_MODULE() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/ |
| D | trt_convert_api.cc | 54 int num_gpus = grappler::GetNumAvailableGPUs(); in NewCluster() local
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 1262 def num_gpus(self): member in Context
2494 def num_gpus(): function
|