Home
last modified time | relevance | path

Searched defs:GetAllocator (Results 1 – 25 of 53) sorted by relevance

123

/external/tensorflow/tensorflow/core/grappler/optimizers/
Devaluation_utils.h44 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_factory.cc44 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/
Dcomposite_device.h39 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in GetAllocator() function
Dsingle_threaded_cpu_device.cc85 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::__anoneb44c9aa0111::SingleThreadedCpuDevice
Ddevice_mgr_test.cc35 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() function in tensorflow::__anon109716260111::CreateDevice::FakeDevice
Drenamed_device.h58 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function
Ddevice_set_test.cc33 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in Dev() function in tensorflow::__anon11a7156c0111::Dev::FakeDevice
Dscoped_allocator_mgr.cc59 ScopedAllocator* ScopedAllocatorContainer::GetAllocator(int32_t scope_id) { in GetAllocator() function in tensorflow::ScopedAllocatorContainer
Dthreadpool_device.cc106 Allocator* ThreadPoolDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::ThreadPoolDevice
/external/intel-media-driver/media_softlet/agnostic/common/shared/bufferMgr/
Dmedia_allocator.h50 static Allocator &GetAllocator(PMOS_INTERFACE osInterface) in GetAllocator() function
/external/pigweed/pw_allocator/public/pw_allocator/
Dblock_allocator_testing.h173 BlockAllocatorTest<BlockAllocatorType, kBufferSize>::GetAllocator() { in GetAllocator() function
180 BlockAllocatorTest<BlockAllocatorType, kBufferSize>::GetAllocator( in GetAllocator() function
/external/tensorflow/tensorflow/core/framework/
Dallocator_registry.cc68 Allocator* AllocatorFactoryRegistry::GetAllocator() { in GetAllocator() function in tensorflow::AllocatorFactoryRegistry
Dresource_op_kernel_test.cc41 Allocator* GetAllocator(AllocatorAttributes) override { in GetAllocator() function in tensorflow::__anone97bdff90111::StubDevice
/external/tensorflow/tensorflow/core/tpu/
Dvirtual_device.cc74 Allocator* VirtualDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::VirtualDevice
/external/tensorflow/tensorflow/core/transforms/utils/
Deval_utils.cc60 tensorflow::Allocator *SimpleDevice::GetAllocator( in GetAllocator() function in mlir::tfg::util::SimpleDevice
/external/intel-media-driver/media_softlet/agnostic/common/vp/hal/pipeline/
Dvp_pipeline.h225 VpAllocator *GetAllocator() in GetAllocator() function
637 VpAllocator& GetAllocator() in GetAllocator() function
/external/tensorflow/tensorflow/compiler/jit/
Dxla_platform_info.cc150 std::shared_ptr<se::DeviceMemoryAllocator> GetAllocator( in GetAllocator() function
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compilation_device.cc82 Allocator* XlaCompilationDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::XlaCompilationDevice
/external/pytorch/c10/core/
DAllocator.cpp51 at::Allocator* GetAllocator(const at::DeviceType& t) { in GetAllocator() function
/external/tensorflow/tensorflow/core/distributed_runtime/
Ddevice_resolver_distributed_test.cc42 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() function in tensorflow::__anon1201900e0111::NewDevice::FakeDevice
Dremote_device.cc43 Allocator* GetAllocator(AllocatorAttributes attr) override { return nullptr; } in GetAllocator() function in tensorflow::RemoteDevice
Dtensor_coding_test.cc41 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::DummyDevice
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dplacement_test.cc61 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() function in tensorflow::__anon1dd2fcbd0111::CreateDevice::FakeDevice
/external/tensorflow/tensorflow/c/kernels/
Dsummary_op_test.cc42 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() function in tensorflow::__anon893a925f0111::DummyDevice
Dbitcast_op_test.cc31 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() function in tensorflow::__anon26d3ac5b0111::DummyDevice

123