/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
D | tf_framework_c_interface.cc | 30 Allocator* GetAllocator(void* op_kernel_ctx) { in GetAllocator() function 67 return GetAllocator(op_kernel_ctx) in _mlir_ciface_tf_alloc() 73 GetAllocator(op_kernel_ctx)->DeallocateRaw(ptr); in _mlir_ciface_tf_dealloc()
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | xla_platform_info.cc | 111 se::DeviceMemoryAllocator* GetAllocator( in GetAllocator() function 123 tf_allocator_adapter->emplace(device->GetAllocator({}), platform); in GetAllocator() 126 tf_allocator_adapter->emplace(device->GetAllocator({}), stream); in GetAllocator() 146 GetAllocator(tf_allocator_adapter, device, stream, platform_info); in GenerateCompilerOptions()
|
D | xla_platform_info.h | 97 se::DeviceMemoryAllocator* GetAllocator(
|
/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
D | gpu_device_factory.cc | 44 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::GPUDevice 95 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::GPUCompatibleCPUDevice 101 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 | 57 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function 58 return underlying_device_->GetAllocator(attr); in GetAllocator()
|
D | threadpool_device.h | 32 Allocator* GetAllocator(AllocatorAttributes attr) override;
|
D | composite_device.h | 39 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in GetAllocator() function
|
D | single_threaded_cpu_device.cc | 85 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function in tensorflow::__anonfe4177f40111::SingleThreadedCpuDevice
|
D | device_mgr_test.cc | 35 Allocator* GetAllocator(AllocatorAttributes) override { return nullptr; } in CreateDevice() function in tensorflow::__anon1a2a6eb00111::CreateDevice::FakeDevice
|
/external/tensorflow/tensorflow/core/kernels/ |
D | ops_testutil.cc | 81 allocator_ = device_->GetAllocator(AllocatorAttributes()); in OpsTestBase() 121 allocator_ = device_->GetAllocator(AllocatorAttributes()); in SetDevice() 127 allocator_ = device_->GetAllocator(AllocatorAttributes()); in SetDevice()
|
D | stack.cc | 247 Allocator* allocator = device->GetAllocator(alloc_attrs); in ComputeAsync() 256 Allocator* cpu_allocator = device->GetAllocator(host_alloc_attrs); in ComputeAsync() 305 Allocator* gpu_allocator = device->GetAllocator(value.alloc_attrs); in ComputeAsync()
|
/external/tensorflow/tensorflow/core/kernels/image/ |
D | resize_area_op_test.cc | 45 Tensor* input = new Tensor(device_->GetAllocator(AllocatorAttributes()), in SetRandomImageInput() 157 new Tensor(device_->GetAllocator(AllocatorAttributes()), in RunRandomTest()
|
/external/tensorflow/tensorflow/core/framework/ |
D | resource_op_kernel_test.cc | 41 Allocator* GetAllocator(AllocatorAttributes) override { in GetAllocator() function in tensorflow::__anon1ebbe4430111::StubDevice 101 DEVICE_CPU, &device_, device_.GetAllocator(AllocatorAttributes()), in CreateOp()
|
D | device_base.h | 181 virtual Allocator* GetAllocator(AllocatorAttributes /*attr*/) { in GetAllocator() function 190 return GetAllocator(attr); in GetStepAllocator()
|
D | allocator.cc | 70 AllocatorFactoryRegistry::singleton()->GetAllocator(); in cpu_allocator_base()
|
/external/tensorflow/tensorflow/core/tpu/ |
D | virtual_device.h | 30 Allocator* GetAllocator(AllocatorAttributes attr) override;
|
/external/tensorflow/tensorflow/compiler/tf2xla/ |
D | xla_compilation_device.h | 53 Allocator* GetAllocator(AllocatorAttributes attr) override;
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | evaluation_utils.h | 44 Allocator* GetAllocator(AllocatorAttributes attr) override { in GetAllocator() function
|
/external/abseil-cpp/absl/container/internal/ |
D | inlined_vector.h | 182 AllocatorTraits::deallocate(GetAllocator(), GetData(), GetCapacity()); in ~AllocationTransaction() 189 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function 195 GetData() = AllocatorTraits::allocate(GetAllocator(), capacity); in Allocate() 222 inlined_vector_internal::DestroyElements(std::addressof(GetAllocator()), in ~ConstructionTransaction() 230 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function 237 inlined_vector_internal::ConstructElements(std::addressof(GetAllocator()), in Construct()
|
/external/webrtc/third_party/abseil-cpp/absl/container/internal/ |
D | inlined_vector.h | 182 AllocatorTraits::deallocate(GetAllocator(), GetData(), GetCapacity()); in ~AllocationTransaction() 189 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function 195 GetData() = AllocatorTraits::allocate(GetAllocator(), capacity); in Allocate() 222 inlined_vector_internal::DestroyElements(std::addressof(GetAllocator()), in ~ConstructionTransaction() 230 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function 237 inlined_vector_internal::ConstructElements(std::addressof(GetAllocator()), in Construct()
|
/external/rust/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/container/internal/ |
D | inlined_vector.h | 182 AllocatorTraits::deallocate(GetAllocator(), GetData(), GetCapacity()); in ~AllocationTransaction() 189 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function 195 GetData() = AllocatorTraits::allocate(GetAllocator(), capacity); in Allocate() 222 inlined_vector_internal::DestroyElements(std::addressof(GetAllocator()), in ~ConstructionTransaction() 230 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function 237 inlined_vector_internal::ConstructElements(std::addressof(GetAllocator()), in Construct()
|
/external/openscreen/third_party/abseil/src/absl/container/internal/ |
D | inlined_vector.h | 182 AllocatorTraits::deallocate(GetAllocator(), GetData(), GetCapacity()); in ~AllocationTransaction() 189 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function 195 GetData() = AllocatorTraits::allocate(GetAllocator(), capacity); in Allocate() 222 inlined_vector_internal::DestroyElements(std::addressof(GetAllocator()), in ~ConstructionTransaction() 230 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function 237 inlined_vector_internal::ConstructElements(std::addressof(GetAllocator()), in Construct()
|
/external/libtextclassifier/abseil-cpp/absl/container/internal/ |
D | inlined_vector.h | 182 AllocatorTraits::deallocate(GetAllocator(), GetData(), GetCapacity()); in ~AllocationTransaction() 189 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function 195 GetData() = AllocatorTraits::allocate(GetAllocator(), capacity); in Allocate() 222 inlined_vector_internal::DestroyElements(std::addressof(GetAllocator()), in ~ConstructionTransaction() 230 AllocatorType& GetAllocator() { return alloc_data_.template get<0>(); } in GetAllocator() function 237 inlined_vector_internal::ConstructElements(std::addressof(GetAllocator()), in Construct()
|
/external/tensorflow/tensorflow/core/kernels/sparse/ |
D | zeros_op.h | 58 Allocator* allocator = c->device()->GetAllocator(AllocatorAttributes()); in operator()
|