HomeSort by relevance Sort by last modified time
    Searched defs:gpu_allocator (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/gpu/
process_state.cc 120 VisitableAllocator* gpu_allocator; local
129 gpu_allocator =
136 gpu_allocator = new GPUDebugAllocator(gpu_allocator, cuda_gpu_id);
137 gpu_allocator = new GPUNanResetAllocator(gpu_allocator, cuda_gpu_id);
142 gpu_allocator = new GPUcudaMallocAllocator(gpu_allocator, cuda_gpu_id);
144 gpu_allocators_[tf_gpu_id.value()] = gpu_allocator;
153 gpu_allocator->AddAllocVisitor(v)
    [all...]
  /external/tensorflow/tensorflow/contrib/nccl/kernels/
nccl_manager_test.cc 64 static Allocator* gpu_allocator(BaseGPUDevice* device) { function in class:tensorflow::NcclManagerTest
140 test_case->ins.emplace_back(gpu_allocator(device), DT_FLOAT, shape);
141 test_case->outs.emplace_back(gpu_allocator(device), DT_FLOAT, shape);
  /external/tensorflow/tensorflow/core/kernels/
stack_ops.cc 415 Allocator* gpu_allocator = device->GetAllocator(value.alloc_attrs); variable
417 new Tensor(gpu_allocator, cpu_tensor->dtype(), cpu_tensor->shape());

Completed in 375 milliseconds