| /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/ |
| 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::__anon38d2dc790111::SingleThreadedCpuDevice
|
| D | device_mgr_test.cc | 35 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() function in tensorflow::__anon08ffc4550111::CreateDevice::FakeDevice
|
| D | renamed_device.h | 58 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::__anon0a0fc39b0111::Dev::FakeDevice
|
| D | scoped_allocator_mgr.cc | 59 ScopedAllocator* ScopedAllocatorContainer::GetAllocator(int32_t scope_id) { in GetAllocator() function in tensorflow::ScopedAllocatorContainer
|
| D | threadpool_device.cc | 106 Allocator* ThreadPoolDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::ThreadPoolDevice
|
| /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::__anon486793c80111::StubDevice
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | virtual_device.cc | 74 Allocator* VirtualDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::VirtualDevice
|
| /external/tensorflow/tensorflow/core/transforms/utils/ |
| D | eval_utils.cc | 60 tensorflow::Allocator *SimpleDevice::GetAllocator( in GetAllocator() function in mlir::tfg::util::SimpleDevice
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_platform_info.cc | 150 std::shared_ptr<se::DeviceMemoryAllocator> GetAllocator( in GetAllocator() function
|
| /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::__anonbc28e37d0111::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 | 37 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::__anone67ae04c0111::CreateDevice::FakeDevice
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | summary_op_test.cc | 42 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() function in tensorflow::__anon9497e9ce0111::DummyDevice
|
| D | bitcast_op_test.cc | 31 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() function in tensorflow::__anon323103ca0111::DummyDevice
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| D | stream_executor_pimpl.h | 537 StreamExecutorMemoryAllocator* GetAllocator() { return &allocator_; } in GetAllocator() function
|
| /external/rust/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/container/internal/ |
| D | inlined_vector.h | 212 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function
253 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function
|
| /external/libtextclassifier/abseil-cpp/absl/container/internal/ |
| D | inlined_vector.h | 189 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function
230 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function
|