| /aosp_15_r20/external/pigweed/pw_allocator/public/pw_allocator/ |
| H A D | pmr_allocator.h | 57 void set_allocator(Allocator& allocator) { allocator_ = &allocator; } in set_allocator() function
81 memory_resource_.set_allocator(allocator); in PmrAllocator()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | local_client_allocation_test.cc | 54 options.set_allocator(allocator); in XLA_TEST_F()
92 ExecutableRunOptions().set_device_ordinal(d).set_allocator(allocator)); in XLA_TEST_F()
|
| H A D | multiple_devices_on_host_test.cc | 44 execute_options.set_allocator( in CompileAndExecute()
|
| H A D | local_client_test_base.cc | 156 run_options.set_allocator(GetOrCreateAllocator(local_client_->platform())); in DefaultExecutableRunOptions()
|
| H A D | local_client_execute_test.cc | 714 DefaultExecutableRunOptions().set_allocator(&allocator)); in XLA_TEST_F()
969 run_options.set_allocator(&allocator).set_stream(stream.get()); in BM_LocalClientOverhead()
|
| H A D | buffer_donation_test.cc | 88 run_options.set_allocator(&memory_allocator); in RunAndCheck()
|
| H A D | xla_hlo_profile_test.cc | 172 exec_run_options.set_allocator(backend->memory_allocator()); in ExecuteAndFetchProfile()
|
| H A D | dynamic_ops_test.cc | 809 options.set_allocator(&allocator); in BM_DynamicSlice()
|
| H A D | exhaustive_op_test_utils.h | 276 run_opts.set_allocator(client_->backend().memory_allocator());
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | StorageUtils.cpp | 53 origStorageImpl->set_allocator(newStorageImpl->allocator()); in share_memory_()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | executable_run_options.cc | 43 ExecutableRunOptions& ExecutableRunOptions::set_allocator( in set_allocator() function in xla::ExecutableRunOptions
|
| H A D | executable_run_options.h | 85 ExecutableRunOptions& set_allocator(
|
| /aosp_15_r20/external/executorch/runtime/executor/ |
| H A D | tensor_parser_aten.cpp | 97 storage->set_allocator(at::getCPUAllocator()); in parseTensor()
|
| /aosp_15_r20/external/pytorch/torch/csrc/ |
| H A D | StorageSharing.cpp | 129 storage.unsafeGetStorageImpl()->set_allocator(new_storage.allocator()); in THPStorage_shareFilename()
227 storage.unsafeGetStorageImpl()->set_allocator(new_storage.allocator()); in THPStorage_shareFd()
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | StorageImpl.h | 193 void set_allocator(at::Allocator* allocator) { in set_allocator() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_compile_on_demand_op.cc | 91 run_options.set_allocator(allocator); in Run()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| H A D | mlir_gpu_test_base.cc | 81 executable_run_options.set_allocator(backend_->memory_allocator()); in RunMlirModule()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Resize.cpp | 311 storage.unsafeGetStorageImpl()->set_allocator( in resize_bytes_nocuda()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_execute.cc | 448 run_options.set_allocator(backend->memory_allocator()); in TPUExecute()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | service.cc | 430 options.set_allocator(backend->memory_allocator()); in ExecuteParallelAndRegisterResult()
528 options.set_allocator(backend->memory_allocator()); in ExecuteAndRegisterResult()
|
| H A D | hlo_runner.cc | 478 run_options.set_allocator(backend().memory_allocator()); in GetServiceRunOptionsForDevice()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tools/ |
| H A D | replay_computation.cc | 314 run_options.set_allocator(&allocator); in ReplayComputation()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/ |
| H A D | local_client.cc | 162 run_options.set_allocator(backend_->memory_allocator()); in RunHelper()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| H A D | xrt_execute_op.cc | 315 run_options.set_allocator(allocator); in RunExecutable()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/kernels/ |
| H A D | xla_ops.cc | 182 run_options.set_allocator(allocator); in RunExecutable()
|