| /external/tensorflow/tensorflow/python/distribute/v1/ |
| D | all_reduce_test.py | 89 def _buildInput(self, num_workers, num_gpus): argument
144 def _buildRing(self, num_workers, num_gpus, subdiv): argument
149 def _testAllReduce(self, num_workers, num_gpus, shape, build_f): argument
165 def _testRingAllReduce(self, num_workers, num_gpus, shape, subdiv): argument
189 def _buildShuffle(self, num_workers, num_gpus, num_shards): argument
196 def _testShuffleAllReduce(self, num_workers, num_gpus, shape, num_shards): argument
216 def _buildRecursiveHD(self, num_workers, num_gpus): argument
222 def _testRecursiveHDAllReduce(self, num_workers, num_gpus, shape): argument
|
| D | cross_device_ops_test.py | 460 num_gpus=0, argument
531 num_gpus, argument
670 num_gpus, argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy_test.py | 76 num_gpus=None, argument
115 def _get_test_objects(self, task_type, task_id, num_gpus): argument
123 def _test_device_assignment_distributed(self, task_type, task_id, num_gpus): argument
223 self, task_type, task_id, num_gpus): argument
274 num_gpus=0): argument
372 def _test_simple_increment(self, task_type, task_id, num_gpus): argument
436 def _test_minimize_loss_graph(self, task_type, task_id, num_gpus): argument
525 num_gpus, argument
607 def testDeviceAssignmentDistributed(self, num_gpus): argument
612 def testDeviceAssignmentDistributedEnablePartitioner(self, num_gpus): argument
[all …]
|
| D | collective_all_reduce_strategy_test.py | 72 num_gpus=None): argument
105 def _get_test_object(self, task_type, task_id, num_gpus=0): argument
113 def _test_minimize_loss_graph(self, task_type, task_id, num_gpus): argument
185 def _test_variable_initialization(self, task_type, task_id, num_gpus): argument
216 num_gpus, argument
|
| D | device_util.py | 130 def local_devices_from_num_gpus(num_gpus): argument
|
| D | central_storage_strategy.py | 74 def _from_num_gpus(cls, num_gpus): argument
|
| D | multi_worker_test_base.py | 533 def _run_client(self, client_fn, task_type, task_id, num_gpus, eager_mode, argument
547 def _run_between_graph_clients(self, client_fn, cluster_spec, num_gpus, *args, argument
|
| D | mirrored_strategy.py | 175 def all_local_devices(num_gpus=None): argument
|
| /external/tensorflow/tensorflow/python/keras/mixed_precision/ |
| D | loss_scale_benchmark.py | 36 def _get_strategy(num_gpus): argument
47 def _benchmark(self, gradient_type, num_gpus, mode, loss_scaling): 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/core/grappler/ |
| D | devices.cc | 46 int num_gpus = gpu_manager->VisibleDeviceCount(); in GetNumAvailableGPUs() local
|
| /external/tensorflow/tensorflow/python/keras/benchmarks/ |
| D | distribution_util.py | 87 num_gpus=0, argument
|
| D | benchmark_util.py | 108 num_gpus=0, argument
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer.cc | 51 int num_gpus = 0; in GetNumGPUs() local
109 const TransposeContext& context, int num_gpus, int num_voltas) { in GetSrcAndDstDataFormats()
412 const int num_gpus = num_gpus_and_num_volta.first; in Optimize() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops_gpu_test.py | 50 def _setup_context(self, num_gpus=2): argument
|
| /external/tensorflow/tensorflow/core/profiler/internal/gpu/ |
| D | cupti_collector.h | 191 uint32 num_gpus; member
201 explicit AnnotationMap(uint64 max_size, uint32 num_gpus) in AnnotationMap()
|
| /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/python/eager/ |
| D | context.py | 1119 def num_gpus(self): member in Context
2219 def num_gpus(): function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_nccl_test.cc | 119 int num_gpus = local_devices.size(); in Init() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | hierarchical_tree_broadcaster_test.cc | 541 int num_gpus) { in PrepColParamsForSubdivPermsTest()
|