| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.h | 44 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_factory.cc | 44 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::GPUDevice
95 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::GPUCompatibleCPUDevice
|
| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_device.cc | 38 Allocator* SYCLDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::SYCLDevice
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | allocator_registry.cc | 68 Allocator* AllocatorFactoryRegistry::GetAllocator() { in GetAllocator() function in tensorflow::AllocatorFactoryRegistry
|
| D | resource_op_kernel_test.cc | 41 Allocator* GetAllocator(AllocatorAttributes) override { in GetAllocator() function in tensorflow::__anond205bfce0111::StubDevice
|
| D | device_base.h | 184 virtual Allocator* GetAllocator(AllocatorAttributes /*attr*/) { in GetAllocator() function
|
| D | resource_mgr_test.cc | 239 Allocator* GetAllocator(AllocatorAttributes) override { in GetAllocator() function in tensorflow::StubDevice
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_cpu_device.cc | 85 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::__anon9ec7ae7f0111::SingleThreadedCpuDevice
|
| D | device_mgr_test.cc | 35 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() function in tensorflow::__anon38effe1b0111::CreateDevice::FakeDevice
|
| D | threadpool_device.cc | 77 Allocator* ThreadPoolDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::ThreadPoolDevice
|
| D | renamed_device.h | 62 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function
|
| D | device_set_test.cc | 33 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in Dev() function in tensorflow::__anon39fffd610111::Dev::FakeDevice
|
| D | scoped_allocator_mgr.cc | 59 ScopedAllocator* ScopedAllocatorContainer::GetAllocator(int32 scope_id) { in GetAllocator() function in tensorflow::ScopedAllocatorContainer
|
| D | dynamic_device_mgr_test.cc | 34 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() function in tensorflow::__anonb07f585f0111::CreateDevice::FakeDevice
|
| D | buf_rendezvous_test.cc | 37 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() function in tensorflow::__anon3d4b4ed70111::BufRendezvousTest::NewDevice::FakeDevice
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.cc | 82 Allocator* XlaCompilationDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::XlaCompilationDevice
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_test.cc | 31 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() function in tensorflow::__anon85d2167e0111::CreateDevice::FakeDevice
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device.cc | 41 Allocator* GetAllocator(AllocatorAttributes attr) override { return nullptr; } in GetAllocator() function in tensorflow::RemoteDevice
|
| D | tensor_coding_test.cc | 41 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::DummyDevice
|
| D | session_mgr_test.cc | 34 Allocator* GetAllocator(AllocatorAttributes attr) override { return nullptr; } in GetAllocator() function in tensorflow::FakeDevice
|
| D | device_resolver_distributed_test.cc | 53 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() function in tensorflow::__anonc9eaa5c30111::NewDevice::FakeDevice
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 32 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() function in tensorflow::__anon7c8b2e100111::DummyDevice
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 518 StreamExecutorMemoryAllocator *GetAllocator() { return &allocator_; } in GetAllocator() function
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 156 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() function in tensorflow::DummyDevice
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 251 Allocator* XlaDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::XlaDevice
|