| /external/tensorflow/tensorflow/core/common_runtime/sycl/ |
| D | sycl_device.cc | 38 Allocator* SYCLDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::SYCLDevice
50 Allocator* host_alloc = GetAllocator(attr); in MakeTensorFromProto()
61 Tensor copy(GetAllocator(alloc_attrs), parsed.dtype(), parsed.shape()); in MakeTensorFromProto()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device_factory.cc | 43 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::GPUDevice
94 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::GPUCompatibleCPUDevice
100 return ThreadPoolDevice::GetAllocator(attr); in GetAllocator()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | scoped_allocator_mgr_test.cc | 149 ScopedAllocator* other = sac->GetAllocator(scope_id_); in TEST_F()
154 other = sac->GetAllocator(scope_id_); in TEST_F()
171 EXPECT_EQ(nullptr, sac->GetAllocator(scope_id_)); in TEST_F()
|
| D | renamed_device.h | 58 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function
59 return underlying_->GetAllocator(attr); in GetAllocator()
|
| D | threadpool_device.h | 32 Allocator* GetAllocator(AllocatorAttributes attr) override;
|
| D | ring_reducer.cc | 114 col_ctx_->device->GetAllocator(attr))); in ContinueAfterInputCopy()
123 group_size_tensor_ = ca_->Scalar(col_ctx_->device->GetAllocator( in ContinueAfterInputCopy()
|
| D | single_threaded_cpu_device.cc | 76 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::__anona485c93d0111::SingleThreadedCpuDevice
|
| D | threadpool_device.cc | 76 Allocator* ThreadPoolDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::ThreadPoolDevice
|
| D | device_set_test.cc | 32 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in Dev() function in tensorflow::__anonf0ab42df0111::Dev::FakeDevice
|
| D | scoped_allocator_mgr.h | 40 ScopedAllocator* GetAllocator(int32 scope_id);
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_base.h | 182 virtual Allocator* GetAllocator(AllocatorAttributes /*attr*/) { in GetAllocator() function
191 return GetAllocator(attr); in GetStepAllocator()
|
| D | resource_op_kernel_test.cc | 41 Allocator* GetAllocator(AllocatorAttributes) override { in GetAllocator() function in tensorflow::__anonb2ebfc8c0111::StubDevice
100 DEVICE_CPU, &device_, device_.GetAllocator(AllocatorAttributes()), in CreateOp()
|
| D | allocator_registry.h | 71 Allocator* GetAllocator();
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | stack.cc | 246 Allocator* allocator = device->GetAllocator(alloc_attrs); in ComputeAsync()
255 Allocator* cpu_allocator = device->GetAllocator(host_alloc_attrs); in ComputeAsync()
304 Allocator* gpu_allocator = device->GetAllocator(value.alloc_attrs); in ComputeAsync()
|
| D | resize_area_op_test.cc | 42 Tensor* input = new Tensor(device_->GetAllocator(AllocatorAttributes()), in SetRandomImageInput()
154 new Tensor(device_->GetAllocator(AllocatorAttributes()), in RunRandomTest()
|
| D | resize_bilinear_op_test.cc | 51 Tensor* input = new Tensor(device_->GetAllocator(AllocatorAttributes()), in SetRandomImageInput()
130 device_->GetAllocator(AllocatorAttributes()), in TestResize()
213 new Tensor(device_->GetAllocator(AllocatorAttributes()), in TEST_F()
|
| D | ops_testutil.cc | 36 allocator_ = device_->GetAllocator(AllocatorAttributes()); in SetDevice()
|
| D | resize_bicubic_op_test.cc | 43 Tensor* input = new Tensor(device_->GetAllocator(AllocatorAttributes()), in SetRandomImageInput()
172 device_->GetAllocator(AllocatorAttributes()), in RunRandomTest()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_compilation_device.h | 53 Allocator* GetAllocator(AllocatorAttributes attr) override;
|
| D | xla_compilation_device.cc | 86 Allocator* XlaCompilationDevice::GetAllocator(AllocatorAttributes attr) { in GetAllocator() function in tensorflow::XlaCompilationDevice
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | evaluation_utils.h | 44 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function
|
| D | evaluation_utils.cc | 93 CreateOpKernel("CPU", cpu_device, cpu_device->GetAllocator({}), node, in EvaluateNode()
|
| /external/tensorflow/tensorflow/contrib/memory_stats/kernels/ |
| D | memory_stats_ops.cc | 26 context->device()->GetAllocator(AllocatorAttributes()); in Compute()
|
| /external/tensorflow/tensorflow/c/kernels/ |
| D | bitcast_op_test.cc | 29 Allocator* GetAllocator(AllocatorAttributes /*attr*/) override { in GetAllocator() function in tensorflow::__anon6753de0e0111::DummyDevice
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 230 auto* allocator = device->GetAllocator(AllocatorAttributes()); in GetPeakMemoryUsage()
458 auto* allocator = device->GetAllocator(AllocatorAttributes()); in ClearAllocatorStats()
|