Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dallocator.h365 struct AllocatorAttributes { struct
372 void Merge(AllocatorAttributes other) { in Merge() argument
384 bool IsEqualOrLessRestrictiveThan(const AllocatorAttributes& other) const { in IsEqualOrLessRestrictiveThan() argument
392 uint32 value = 0;
395 int32 scope_id = 0;
Dresource_op_kernel_test.cc41 Allocator* GetAllocator(AllocatorAttributes) override { in GetAllocator() argument
Ddevice_base.h181 virtual Allocator* GetAllocator(AllocatorAttributes /*attr*/) { in GetAllocator() argument
Dresource_mgr_test.cc241 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/
Dcomposite_device.h39 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in GetAllocator() argument
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.cc39 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
Dbuf_rendezvous_test.cc39 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() argument
/external/tensorflow/tensorflow/core/distributed_runtime/
Ddevice_resolver_distributed_test.cc42 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() argument
Dcollective_param_resolver_distributed_test.cc40 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/core/common_runtime/eager/
Dplacement_test.cc61 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
Dcontext_test.cc35 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
Dtensor_handle_test.cc76 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
/external/tensorflow/tensorflow/c/kernels/
Dsummary_op_test.cc42 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() argument
Dbitcast_op_test.cc31 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() argument
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Drpc_rendezvous_mgr_test.cc103 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
/external/tensorflow/tensorflow/c/
Dkernels_test.cc497 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() argument