Searched refs:BaseGPUDevice (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 231 class BaseGPUDevice::StreamGroupFactory {
237 BaseGPUDevice::StreamGroup* GetOrCreate(TfGpuId tf_gpu_id, in GetOrCreate()
298 BaseGPUDevice::BaseGPUDevice(const SessionOptions& options, const string& name, in BaseGPUDevice() function in tensorflow::BaseGPUDevice
334 BaseGPUDevice::~BaseGPUDevice() { in ~BaseGPUDevice()
341 Status BaseGPUDevice::InitScratchBuffers() { in InitScratchBuffers()
372 Status BaseGPUDevice::Init(const SessionOptions& options) { in Init()
447 bool BaseGPUDevice::RequiresRecordingAccessedTensors() const { in RequiresRecordingAccessedTensors()
454 Status BaseGPUDevice::FillContextMap(const Graph* graph, in FillContextMap()
489 void BaseGPUDevice::Compute(OpKernel* op_kernel, OpKernelContext* context) { in Compute()
507 string BaseGPUDevice::ComputeOpKernelDebugString(const OpKernel& op_kernel, in ComputeOpKernelDebugString()
[all …]
|
| D | gpu_device_factory.cc | 28 class GPUDevice : public BaseGPUDevice {
34 : BaseGPUDevice(options, name, memory_limit, locality, tf_gpu_id, in GPUDevice()
63 std::unique_ptr<BaseGPUDevice> CreateGPUDevice( in CreateGPUDevice()
|
| D | gpu_device.h | 52 class BaseGPUDevice : public LocalDevice {
54 BaseGPUDevice(const SessionOptions& options, const string& name,
60 ~BaseGPUDevice() override;
305 virtual std::unique_ptr<BaseGPUDevice> CreateGPUDevice(
|
| /external/tensorflow/tensorflow/core/nccl/ |
| D | nccl_manager_test.cc | 31 static std::vector<std::unique_ptr<BaseGPUDevice>> GetGPUDevices() { in GetGPUDevices()
40 std::vector<std::unique_ptr<BaseGPUDevice>> gpus; in GetGPUDevices()
45 gpus.emplace_back(static_cast<BaseGPUDevice*>(device.release())); in GetGPUDevices()
68 devices_ = new std::vector<std::unique_ptr<BaseGPUDevice>>(GetGPUDevices()); in SetUpTestCase()
223 static BaseGPUDevice* GetDevice(size_t rank) { in GetDevice()
228 static Allocator* GpuAllocator(BaseGPUDevice* device) { in GpuAllocator()
239 static std::vector<std::unique_ptr<BaseGPUDevice>>* devices_;
245 std::vector<std::unique_ptr<BaseGPUDevice>>* NcclManagerTest<Scalar>::devices_ =
|