Home
last modified time | relevance | path

Searched refs:BaseGPUDevice (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_test.cc65 BaseGPUDevice::TestOnlyReset(); in TearDown()
205 EXPECT_EQ(0, static_cast<BaseGPUDevice*>(devices[0].get())->priority()); in TEST_F()
217 EXPECT_EQ(0, static_cast<BaseGPUDevice*>(devices[0].get())->priority()); in TEST_F()
227 EXPECT_EQ(0, static_cast<BaseGPUDevice*>(devices[0].get())->priority()); in TEST_F()
288 EXPECT_EQ(0, static_cast<BaseGPUDevice*>(devices[0].get())->priority()); in TEST_F()
301 EXPECT_EQ(0, static_cast<BaseGPUDevice*>(devices[0].get())->priority()); in TEST_F()
302 EXPECT_EQ(-1, static_cast<BaseGPUDevice*>(devices[1].get())->priority()); in TEST_F()
344 EXPECT_EQ(-1, static_cast<BaseGPUDevice*>(devices[0].get())->priority()); in TEST_F()
345 EXPECT_EQ(0, static_cast<BaseGPUDevice*>(devices[1].get())->priority()); in TEST_F()
Dgpu_device_factory.cc29 class GPUDevice : public BaseGPUDevice {
35 : BaseGPUDevice(options, name, memory_limit, locality, tf_gpu_id, in GPUDevice()
64 std::unique_ptr<BaseGPUDevice> CreateGPUDevice( in CreateGPUDevice()
Dgpu_device.cc233 class BaseGPUDevice::StreamGroupFactory {
239 BaseGPUDevice::StreamGroup* GetOrCreate(TfGpuId tf_gpu_id, in GetOrCreate()
379 BaseGPUDevice::BaseGPUDevice(const SessionOptions& options, const string& name, in BaseGPUDevice() function in tensorflow::BaseGPUDevice
396 BaseGPUDevice::~BaseGPUDevice() { in ~BaseGPUDevice()
403 Status BaseGPUDevice::InitScratchBuffers() { in InitScratchBuffers()
424 Status BaseGPUDevice::Init(const SessionOptions& options) { in Init()
531 string BaseGPUDevice::ComputeOpKernelDebugString(const OpKernel& op_kernel, in ComputeOpKernelDebugString()
538 void BaseGPUDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
603 Status BaseGPUDevice::Sync() { in Sync()
615 void BaseGPUDevice::ComputeAsync(AsyncOpKernel* op_kernel, in ComputeAsync()
[all …]
Dgpu_device.h52 class BaseGPUDevice : public LocalDevice {
54 BaseGPUDevice(const SessionOptions& options, const std::string& name,
60 ~BaseGPUDevice() override;
360 virtual std::unique_ptr<BaseGPUDevice> CreateGPUDevice(
/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager_test.cc34 static std::vector<std::unique_ptr<BaseGPUDevice>> GetGPUDevices() { in GetGPUDevices()
38 std::vector<std::unique_ptr<BaseGPUDevice>> gpus; in GetGPUDevices()
43 gpus.emplace_back(static_cast<BaseGPUDevice*>(device.release())); in GetGPUDevices()
71 devices_ = new std::vector<std::unique_ptr<BaseGPUDevice>>(GetGPUDevices()); in SetUpTestSuite()
425 static BaseGPUDevice* GetDevice(int num_ranks_per_node, int node, in GetDevice()
435 static Allocator* GpuAllocator(BaseGPUDevice* device) { in GpuAllocator()
445 static std::vector<std::unique_ptr<BaseGPUDevice>>* devices_;
451 std::vector<std::unique_ptr<BaseGPUDevice>>* NcclManagerTest<Scalar>::devices_ =
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr_test.cc150 gpu_.reset(reinterpret_cast<BaseGPUDevice*>(device_.release())); in GPUDeviceTestHelper()
156 BaseGPUDevice* gpu() { return gpu_.get(); } in gpu()
168 std::unique_ptr<BaseGPUDevice> gpu_;