| /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 | 43 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::GPUDevice
94 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::GPUCompatibleCPUDevice
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_cpu_device.cc | 76 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::__anonfe4177f40111::SingleThreadedCpuDevice
|
| D | threadpool_device.cc | 76 Allocator* ThreadPoolDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::ThreadPoolDevice
|
| D | device_set_test.cc | 32 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in Dev() function in tensorflow::__anon1b3a6df60111::Dev::FakeDevice
|
| D | renamed_device.h | 58 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function
|
| D | scoped_allocator_mgr.cc | 59 ScopedAllocator* ScopedAllocatorContainer::GetAllocator(int32 scope_id) { in GetAllocator() function in tensorflow::ScopedAllocatorContainer
|
| D | constant_folding_test.cc | 104 Allocator* GetAllocator(AllocatorAttributes attr) override { return nullptr; } in GetAllocator() function in tensorflow::__anon5f0c02c70111::FakeDevice
|
| /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::__anon1ebbe4430111::StubDevice
|
| D | device_base.h | 182 virtual Allocator* GetAllocator(AllocatorAttributes /*attr*/) { in GetAllocator() function
|
| D | resource_mgr_test.cc | 235 Allocator* GetAllocator(AllocatorAttributes) override { in GetAllocator() function in tensorflow::StubDevice
|
| D | op_kernel_test.cc | 313 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() function in tensorflow::__anon843475bc0111::DummyDevice
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | remote_device.cc | 49 Allocator* GetAllocator(AllocatorAttributes attr) override { return nullptr; } in GetAllocator() function in tensorflow::RemoteDevice
|
| D | session_mgr_test.cc | 34 Allocator* GetAllocator(AllocatorAttributes attr) override { return nullptr; } in GetAllocator() function in tensorflow::FakeDevice
|
| D | tensor_coding_test.cc | 41 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::DummyDevice
|
| D | device_resolver_distributed_test.cc | 51 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() function in tensorflow::__anon5f4e7c180111::NewDevice::FakeDevice
|
| D | collective_param_resolver_distributed_test.cc | 38 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() function in tensorflow::__anon53767e020111::NewDevice::FakeDevice
|
| D | collective_rma_distributed_test.cc | 50 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() function in tensorflow::__anond64f02c00111::NewDevice::FakeDevice
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 29 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() function in tensorflow::__anon7be2f0650111::DummyDevice
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.cc | 86 Allocator* XlaCompilationDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::XlaCompilationDevice
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_test.cc | 132 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() function in tensorflow::DummyDevice
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 240 Allocator* XlaDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::XlaDevice
|
| D | increase_dynamism_for_auto_jit_pass_test.cc | 49 Allocator* GetAllocator(AllocatorAttributes attr) override { return nullptr; } in GetAllocator() function in tensorflow::__anonbd6aeead0111::FakeDevice
|