OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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