| /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 | 153 run_options.set_allocator(GetOrCreateAllocator(local_client_->platform())); in DefaultExecutableRunOptions()
|
| D | xla_hlo_profile_test.cc | 172 exec_run_options.set_allocator(backend->memory_allocator()); in ExecuteAndFetchProfile()
|
| D | local_client_execute_test.cc | 705 DefaultExecutableRunOptions().set_allocator(&allocator)); in XLA_TEST_F()
936 run_options.set_allocator(&allocator).set_stream(stream.get()); in BM_LocalClientOverhead()
|
| D | exhaustive_op_test.cc | 304 run_opts.set_allocator(client_->backend().memory_allocator()); in BuildAndRunComputation()
|
| D | dynamic_ops_test.cc | 788 options.set_allocator(&allocator); in BM_DynamicSlice()
|
| D | fusion_test.cc | 904 options.set_allocator(&allocator).set_stream(&stream); in BM_ParallelFusion()
|
| D | while_test.cc | 1323 options.set_allocator(&allocator); in BM_WhileLoop()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.cc | 28 ExecutableRunOptions& ExecutableRunOptions::set_allocator( in set_allocator() function in xla::ExecutableRunOptions
|
| D | executable_run_options.h | 42 ExecutableRunOptions& set_allocator(DeviceMemoryAllocator* allocator);
|
| /external/libxcam/xcore/ |
| D | image_handler.cpp | 38 ImageHandler::set_allocator (const SmartPtr<BufferPool> &allocator) in set_allocator() function in XCam::ImageHandler
|
| D | image_handler.h | 98 bool set_allocator (const SmartPtr<BufferPool> &allocator);
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | buffer_comparator.cc | 144 run_options.set_allocator(allocator_); in CompareEqualImpl()
|
| /external/tensorflow/tensorflow/compiler/jit/kernels/ |
| D | xla_ops.cc | 362 run_options.set_allocator(platform_info_.allocator()); in Compute()
536 run_options.set_allocator(platform_info_.allocator()); in Compute()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_compile_on_demand_op.cc | 77 run_options.set_allocator(client->backend().memory_allocator()); in Run()
|
| /external/libxcam/modules/soft/ |
| D | soft_handler.cpp | 129 set_allocator (new SoftVideoBufAllocator); in confirm_configured()
|
| /external/tensorflow/tensorflow/compiler/xrt/kernels/ |
| D | xrt_execute_op.cc | 209 run_options.set_allocator(device_ref.backend()->memory_allocator()); in DoWork()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | local_client.cc | 160 run_options.set_allocator(backend_->memory_allocator()); in Run()
|
| /external/tensorflow/tensorflow/compiler/xla/tools/ |
| D | replay_computation.cc | 288 run_options.set_allocator(&allocator); in ReplayComputation()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 247 options.set_allocator(client_->backend().memory_allocator()); in Execute()
295 options.set_allocator(client_->backend().memory_allocator()); in ExecutePerReplica()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | service.cc | 450 options.set_allocator(backend->memory_allocator()); in ExecuteParallelAndRegisterResult()
545 options.set_allocator(backend->memory_allocator()); in ExecuteAndRegisterResult()
|
| D | hlo_runner.cc | 447 run_options.set_allocator(backend().memory_allocator()); in GetServiceRunOptionsForDevice()
|