Searched refs:GetScopedAllocator (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | renamed_device.h | 61 Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function
63 return underlying_device_->GetScopedAllocator(attr, step_id); in GetScopedAllocator()
|
| D | threadpool_device.h | 33 Allocator* GetScopedAllocator(AllocatorAttributes attr,
|
| D | threadpool_device.cc | 77 Allocator* ThreadPoolDevice::GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function in tensorflow::ThreadPoolDevice
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 195 virtual Allocator* GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function
|
| D | op_kernel_test.cc | 401 Allocator* GetScopedAllocator(AllocatorAttributes attrs, in GetScopedAllocator() function in tensorflow::__anon843475bc0111::ScopedAllocatorDevice
|
| D | op_kernel.cc | 366 allocator = params_->device->GetScopedAllocator(attr, step_id()); in get_allocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.h | 99 Allocator* GetScopedAllocator(AllocatorAttributes attr,
|
| D | gpu_device.cc | 1066 Allocator* BaseGPUDevice::GetScopedAllocator(AllocatorAttributes attr, in GetScopedAllocator() function in tensorflow::BaseGPUDevice
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scoped_allocator_ops_test.cc | 60 allocator = device_->GetScopedAllocator(attr, context_->step_id()); in MakeOp()
|