Searched defs:cpu_allocator_ (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ | ||
| H A D | pluggable_device.h | 81 Allocator* cpu_allocator_; variable |
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ | ||
| H A D | cpu_allocator_impl.cc | 213 CPUAllocator* cpu_allocator_; member in tensorflow::__anon2523aecd0111::CPUAllocatorFactory::CPUSubAllocator |
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ | ||
| H A D | gpu_device.h | 153 Allocator* cpu_allocator_; // not owned variable |
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ | ||
| H A D | light_outside_compilation.cc | 362 Allocator* cpu_allocator_; member in tensorflow::__anonbb96d3be0311::TfCallbackDevice |