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

  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device_factory.cc 27 class GPUDevice : public BaseGPUDevice {
33 : BaseGPUDevice(options, name, memory_limit, locality, tf_gpu_id,
61 BaseGPUDevice* CreateGPUDevice(const SessionOptions& options,
gpu_device.h 48 class BaseGPUDevice : public LocalDevice {
50 BaseGPUDevice(const SessionOptions& options, const string& name,
56 ~BaseGPUDevice() override;
177 // Creates a BaseGPUDevice associated with 'tf_gpu_id', allocates (strictly)
185 virtual BaseGPUDevice* CreateGPUDevice(const SessionOptions& options,
gpu_device.cc 195 class BaseGPUDevice::StreamGroupFactory {
201 BaseGPUDevice::StreamGroup* GetOrCreate(TfGpuId tf_gpu_id,
249 BaseGPUDevice::BaseGPUDevice(const SessionOptions& options, const string& name,
266 BaseGPUDevice::~BaseGPUDevice() {
271 Status BaseGPUDevice::Init(const SessionOptions& options) {
365 bool BaseGPUDevice::RequiresRecordingAccessedTensors() const {
372 Status BaseGPUDevice::FillContextMap(const Graph* graph,
407 void BaseGPUDevice::Compute(OpKernel* op_kernel, OpKernelContext* context)
    [all...]
  /external/tensorflow/tensorflow/contrib/nccl/kernels/
nccl_manager_test.cc 31 static std::vector<BaseGPUDevice*> GetGPUDevices() {
40 std::vector<BaseGPUDevice*> gpus;
43 gpus.push_back(static_cast<BaseGPUDevice*>(d));
55 devices = new std::vector<BaseGPUDevice*>(GetGPUDevices());
64 static Allocator* gpu_allocator(BaseGPUDevice* device) {
69 static std::vector<BaseGPUDevice*>* devices;
184 std::vector<BaseGPUDevice*>* NcclManagerTest::devices = nullptr;

Completed in 123 milliseconds