| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | allocator.h | 313 struct AllocatorAttributes { struct
320 void Merge(AllocatorAttributes other) { in Merge() argument
332 bool IsEqualOrLessRestrictiveThan(const AllocatorAttributes& other) const { in IsEqualOrLessRestrictiveThan() argument
340 uint32 value = 0;
343 int32 scope_id = 0;
|
| H A D | resource_op_kernel_test.cc | 41 Allocator* GetAllocator(AllocatorAttributes) override { in GetAllocator() argument
|
| H A D | device_base.h | 185 virtual Allocator* GetAllocator(AllocatorAttributes /*attr*/) { in GetAllocator() argument
|
| H A D | resource_mgr_test.cc | 327 Allocator* GetAllocator(AllocatorAttributes) override { in GetAllocator() argument
|
| H A D | op_kernel_test.cc | 355 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | composite_device.h | 39 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in GetAllocator() argument
|
| H A D | device_mgr_test.cc | 35 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
|
| H A D | device_set_test.cc | 33 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in Dev() argument
|
| H A D | dynamic_device_mgr_test.cc | 39 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
|
| H A D | buf_rendezvous_test.cc | 39 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | device_resolver_distributed_test.cc | 42 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() argument
|
| H A D | collective_param_resolver_distributed_test.cc | 43 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in NewDevice() argument
|
| H A D | collective_rma_distributed_test.cc | 64 Allocator* GetAllocator(AllocatorAttributes) override { return allocator_; } in NewDevice() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/eager/ |
| H A D | placement_test.cc | 61 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
|
| H A D | context_test.cc | 39 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
|
| H A D | tensor_handle_test.cc | 76 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/kernels/ |
| H A D | summary_op_test.cc | 42 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() argument
|
| H A D | bitcast_op_test.cc | 31 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | rpc_rendezvous_mgr_test.cc | 107 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/cpp_tests/core_runtime/ |
| H A D | op_handler_selector_test.cc | 99 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | kernels_test.cc | 740 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() argument
|