Home
last modified time | relevance | path

Searched refs:num_numa_nodes (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dthreadpool_device_factory.cc34 int num_numa_nodes = port::NUMANumNodes(); in CreateDevices() local
44 int numa_node = i % num_numa_nodes; in CreateDevices()
46 LOG(INFO) << "Only " << num_numa_nodes in CreateDevices()
Dlocal_device.cc137 int num_numa_nodes = port::NUMANumNodes(); in LocalDevice() local
138 DCHECK_LT(numa_node, num_numa_nodes); in LocalDevice()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dnuma_map_and_batch_dataset_op.cc871 int num_numa_nodes = port::NUMANumNodes(); in EnsureBackgroundThreadsStarted() local
872 if (num_numa_nodes < 1) { in EnsureBackgroundThreadsStarted()
874 num_numa_nodes); in EnsureBackgroundThreadsStarted()
879 VLOG(3) << "# NUMA Nodes: " << num_numa_nodes in EnsureBackgroundThreadsStarted()
881 workers_.resize(num_numa_nodes); in EnsureBackgroundThreadsStarted()
883 num_numa_nodes = workers_.size(); in EnsureBackgroundThreadsStarted()
888 std::max(1LL, (num_parallel_calls_->value + num_numa_nodes - 1) / in EnsureBackgroundThreadsStarted()
889 num_numa_nodes); in EnsureBackgroundThreadsStarted()
891 VLOG(3) << "Starting " << num_threads_per_block * num_numa_nodes in EnsureBackgroundThreadsStarted()
898 for (int i = 0; i < num_numa_nodes; ++i) { in EnsureBackgroundThreadsStarted()
[all …]
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_factory.cc119 int num_numa_nodes = options.config.experimental().use_numa_affinity() in CreateDevices() local
124 int numa_node = i % num_numa_nodes; in CreateDevices()