| /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()
|