| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_process_state.cc | 74 int numa_node = se->GetDeviceDescription().numa_node(); in BusIdForGPU() local
180 Allocator* GPUProcessState::GetGpuHostAllocator(int numa_node) { in GetGpuHostAllocator()
292 int numa_node, const SubAllocator::Visitor& visitor) { in AddGpuHostAllocVisitor()
306 int numa_node, const SubAllocator::Visitor& visitor) { in AddGpuHostFreeVisitor()
|
| D | gpu_host_allocator.h | 29 explicit GpuHostAllocator(se::StreamExecutor* stream_exec, int numa_node, in GpuHostAllocator()
|
| D | pool_allocator_test.cc | 132 [&alloc_count, &alloc_size](void* ptr, int numa_node, int64 size) { in TEST()
139 [&free_count, &free_size](void* ptr, int numa_node, int64 size) { in TEST()
|
| D | gpu_device_factory.cc | 124 int numa_node = i % num_numa_nodes; in CreateDevices() local
|
| D | gpu_device.cc | 699 int numa_node = attributes().locality().numa_node(); in MakeTensorFromProto() local
1193 int numa_node = dev_locality.numa_node(); in CreateGPUDevice() local
1293 int numa_node = desc.numa_node(); in GetDeviceLocalities() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | threadpool_device_factory.cc | 44 int numa_node = i % num_numa_nodes; in CreateDevices() local
|
| D | threadpool_device.cc | 114 virtual SubAllocator* CreateSubAllocator(int numa_node) { in CreateSubAllocator()
|
| D | process_state.cc | 62 Allocator* ProcessState::GetCPUAllocator(int numa_node) { in GetCPUAllocator()
|
| D | pool_allocator.h | 151 BasicCPUAllocator(int numa_node, const std::vector<Visitor>& alloc_visitors, in BasicCPUAllocator()
|
| D | local_device.cc | 136 int numa_node = attributes.locality().numa_node(); in LocalDevice() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | make_numa_aware.cc | 52 auto* numa_node = graph.AddNode(MakeNumaAwareNode(node, &graph)); in OptimizeAndCollectStats() local
|
| /external/tensorflow/tensorflow/contrib/verbs/ |
| D | rdma_mgr.cc | 271 SubAllocator::Visitor free_visitor = [](void* ptr, int numa_node, in RegMemVisitors()
272 size_t num_bytes) { in RegMemVisitors()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_attributes.proto | 26 int32 numa_node = 2; field
|
| D | allocator_registry.cc | 90 SubAllocator* AllocatorFactoryRegistry::GetSubAllocator(int numa_node) { in GetSubAllocator()
|
| D | allocator.cc | 185 SubAllocator* CreateSubAllocator(int numa_node) override { in CreateSubAllocator()
224 Allocator* cpu_allocator(int numa_node) { in cpu_allocator()
|
| /external/tensorflow/tensorflow/c/ |
| D | env.h | 56 int numa_node; member
|
| /external/tensorflow/tensorflow/contrib/gdr/ |
| D | gdr_memory_manager.cc | 242 SubAllocator::Visitor free_visitor = [this](void* ptr, int numa_node, in Init()
243 size_t num_bytes) { in Init()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | constant_op.cc | 265 int numa_node = DeviceNumaNode(ctx->device()); in Compute() local
|
| D | cwise_ops_common.h | 267 int numa_node = DeviceNumaNode(ctx->device()); in Compute() local
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | env.h | 425 int numa_node = port::kNUMANoAffinity; member
|
| /external/tensorflow/tensorflow/core/platform/posix/ |
| D | port.cc | 241 hwloc_obj_t numa_node = GetHWLocTypeIndex(HWLOC_OBJ_NUMANODE, node); in NUMAMalloc() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | device_resolver_distributed_test.cc | 46 int numa_node) { in NewDevice()
|
| /external/kernel-headers/original/uapi/rdma/hfi/ |
| D | hfi1_ioctl.h | 90 __u16 numa_node; /* NUMA node of the assigned device */ member
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_description.h | 65 int numa_node() const { return numa_node_; } in numa_node() function
|
| /external/strace/ |
| D | bpf_attr.h | 65 uint32_t numa_node; member
|