| /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/sycl/ |
| D | sycl_device.cc | 38 Allocator* SYCLDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::SYCLDevice
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_cpu_device.cc | 76 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::__anona485c93d0111::SingleThreadedCpuDevice
|
| D | device_set_test.cc | 32 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in Dev() function in tensorflow::__anonf0ab42df0111::Dev::FakeDevice
|
| D | threadpool_device.cc | 76 Allocator* ThreadPoolDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::ThreadPoolDevice
|
| 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::__anond28d22700111::FakeDevice
|
| /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::__anonb2ebfc8c0111::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::__anon74460fe50111::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::__anon0903cdc10111::NewDevice::FakeDevice
|
| D | collective_param_resolver_distributed_test.cc | 38 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() function in tensorflow::__anonc2f8b6eb0111::NewDevice::FakeDevice
|
| D | collective_rma_distributed_test.cc | 50 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() function in tensorflow::__anon7e5c43490111::NewDevice::FakeDevice
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 29 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() function in tensorflow::__anon6753de0e0111::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::__anon624fd3d60111::FakeDevice
|