| /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/ |
| D | composite_device.h | 39 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in GetAllocator() function
|
| D | single_threaded_cpu_device.cc | 85 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::__anon6eac52db0111::SingleThreadedCpuDevice
|
| D | device_mgr_test.cc | 35 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() function in tensorflow::__anon7c4d83f70111::CreateDevice::FakeDevice
|
| D | threadpool_device.cc | 73 Allocator* ThreadPoolDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::ThreadPoolDevice
|
| D | renamed_device.h | 57 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::__anon7d5d833d0111::Dev::FakeDevice
|
| D | scoped_allocator_mgr.cc | 59 ScopedAllocator* ScopedAllocatorContainer::GetAllocator(int32 scope_id) { in GetAllocator() function in tensorflow::ScopedAllocatorContainer
|
| /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/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::__anon7da9c82a0111::StubDevice
|
| D | device_base.h | 181 virtual Allocator* GetAllocator(AllocatorAttributes /*attr*/) { in GetAllocator() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| D | tf_framework_c_interface.cc | 30 Allocator* GetAllocator(void* op_kernel_ctx) { in GetAllocator() function
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_platform_info.cc | 111 se::DeviceMemoryAllocator* GetAllocator( in GetAllocator() function
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 74 Allocator* VirtualDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::VirtualDevice
|
| /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/distributed_runtime/ |
| D | device_resolver_distributed_test.cc | 42 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() function in tensorflow::__anone296589f0111::NewDevice::FakeDevice
|
| D | remote_device.cc | 43 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
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | placement_test.cc | 61 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() function in tensorflow::__anon751bb52e0111::CreateDevice::FakeDevice
|
| D | context_test.cc | 35 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() function in tensorflow::__anon31761eda0111::CreateDevice::FakeDevice
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | summary_op_test.cc | 42 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() function in tensorflow::__anonaf2956f00111::DummyDevice
|
| D | bitcast_op_test.cc | 31 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() function in tensorflow::__anon4cc270ec0111::DummyDevice
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.h | 529 StreamExecutorMemoryAllocator *GetAllocator() { return &allocator_; } in GetAllocator() function
|