Home
last modified time | relevance | path

Searched defs:AllocatorAttributes (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dallocator.h306 struct AllocatorAttributes { struct
313 void Merge(AllocatorAttributes other) { in Merge() argument
325 bool IsEqualOrLessRestrictiveThan(const AllocatorAttributes& other) const { in IsEqualOrLessRestrictiveThan() argument
333 uint32 value = 0;
336 int32 scope_id = 0;
Dresource_op_kernel_test.cc41 Allocator* GetAllocator(AllocatorAttributes) override { in GetAllocator() argument
Ddevice_base.h184 virtual Allocator* GetAllocator(AllocatorAttributes /*attr*/) { in GetAllocator() argument
Dresource_mgr_test.cc239 Allocator* GetAllocator(AllocatorAttributes) override { in GetAllocator() argument
Dop_kernel_test.cc352 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() argument
/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_mgr_test.cc35 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
Ddevice_set_test.cc33 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in Dev() argument
Ddynamic_device_mgr_test.cc34 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
Dbuf_rendezvous_test.cc37 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() argument
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dcontext_test.cc31 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op_test.cc32 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() argument
/external/tensorflow/tensorflow/core/distributed_runtime/
Ddevice_resolver_distributed_test.cc53 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() argument
Dcollective_param_resolver_distributed_test.cc38 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() argument
Dcollective_rma_distributed_test.cc51 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() argument
/external/tensorflow/tensorflow/c/
Dkernels_test.cc156 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() argument