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