| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | local_client_allocation_test.cc | 55 options.set_allocator(allocator); in XLA_TEST_F()
93 ExecutableRunOptions().set_device_ordinal(d).set_allocator(allocator)); in XLA_TEST_F()
|
| D | multiple_devices_on_host_test.cc | 44 execute_options.set_allocator( in CompileAndExecute()
|
| D | local_client_test_base.cc | 159 run_options.set_allocator(GetOrCreateAllocator(local_client_->platform())); in DefaultExecutableRunOptions()
|
| D | buffer_donation_test.cc | 86 run_options.set_allocator(&memory_allocator); in RunAndCheck()
|
| D | xla_hlo_profile_test.cc | 172 exec_run_options.set_allocator(backend->memory_allocator()); in ExecuteAndFetchProfile()
|
| D | local_client_execute_test.cc | 708 DefaultExecutableRunOptions().set_allocator(&allocator)); in XLA_TEST_F()
984 run_options.set_allocator(&allocator).set_stream(stream.get()); in BM_LocalClientOverhead()
|
| D | dynamic_ops_test.cc | 806 options.set_allocator(&allocator); in BM_DynamicSlice()
|
| D | exhaustive_op_test_utils.h | 270 run_opts.set_allocator(client_->backend().memory_allocator());
|
| D | cpu_gpu_fusion_test.cc | 903 options.set_allocator(&allocator).set_stream(&stream); in BM_ParallelFusion()
|
| D | while_test.cc | 1323 options.set_allocator(&allocator); in BM_WhileLoop()
|
| D | dot_operation_test.cc | 1857 options.set_allocator(&allocator); in DOT_ReorderContracting()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.cc | 43 ExecutableRunOptions& ExecutableRunOptions::set_allocator( in set_allocator() function in xla::ExecutableRunOptions
|
| D | executable_run_options.h | 85 ExecutableRunOptions& set_allocator(
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 83 run_options.set_allocator(allocator); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | mlir_gpu_test_base.cc | 88 executable_run_options.set_allocator(backend_->memory_allocator()); in RunMlirModule()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 273 run_options.set_allocator(allocator); in Compute()
509 run_options.set_allocator(allocator); in Compute()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.cc | 444 run_options.set_allocator(backend->memory_allocator()); in TPUExecute()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 328 run_options.set_allocator(&allocator); in ReplayComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 159 run_options.set_allocator(backend_->memory_allocator()); in RunHelper()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_runner.cc | 448 run_options.set_allocator(backend().memory_allocator()); in GetServiceRunOptionsForDevice()
|
| D | service.cc | 400 options.set_allocator(backend->memory_allocator()); in ExecuteParallelAndRegisterResult()
497 options.set_allocator(backend->memory_allocator()); in ExecuteAndRegisterResult()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 313 run_options.set_allocator(device_ref->backend()->memory_allocator()); in RunExecutable()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 1751 run_options.set_allocator(client_->allocator()); in EnqueueExecution()
|