HomeSort by relevance Sort by last modified time
    Searched refs:set_numa_node (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
threadpool_device_factory.cc 52 dev_locality.set_numa_node(numa_node);
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device_factory.cc 126 locality.set_numa_node(numa_node);
gpu_device.cc     [all...]
  /external/tensorflow/tensorflow/stream_executor/
device_description.h 305 void set_numa_node(int value) { device_description_->numa_node_ = value; } function in class:stream_executor::internal::DeviceDescriptionBuilder
  /external/tensorflow/tensorflow/core/distributed_runtime/
device_resolver_distributed_test.cc 56 attr.mutable_locality()->set_numa_node(numa_node);
collective_rma_distributed_test.cc 55 attr.mutable_locality()->set_numa_node(3); // a non-default value
collective_param_resolver_distributed_test.cc 43 attr.mutable_locality()->set_numa_node(3); // a non-default value
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_gpu_executor.cc     [all...]
  /external/tensorflow/tensorflow/stream_executor/rocm/
rocm_gpu_executor.cc     [all...]

Completed in 702 milliseconds