Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dallocator.h359 struct AllocatorAttributes { struct
366 void Merge(AllocatorAttributes other) { in Merge() argument
374 bool IsEqualOrLessRestrictiveThan(const AllocatorAttributes& other) const { in IsEqualOrLessRestrictiveThan() argument
382 uint32 value = 0;
385 int32 scope_id = 0;
Dresource_op_kernel_test.cc41 Allocator* GetAllocator(AllocatorAttributes) override { in GetAllocator() argument
Ddevice_base.h182 virtual Allocator* GetAllocator(AllocatorAttributes /*attr*/) { in GetAllocator() argument
Dresource_mgr_test.cc235 Allocator* GetAllocator(AllocatorAttributes) override { in GetAllocator() argument
Dop_kernel_test.cc313 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() argument
/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_set_test.cc32 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in Dev() argument
/external/tensorflow/tensorflow/c/kernels/
Dbitcast_op_test.cc29 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() argument
/external/tensorflow/tensorflow/c/
Dkernels_test.cc132 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() argument
/external/tensorflow/tensorflow/core/distributed_runtime/
Ddevice_resolver_distributed_test.cc51 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.cc50 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() argument