Home
last modified time | relevance | path

Searched refs:set_allocator (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dlocal_client_allocation_test.cc55 options.set_allocator(allocator); in XLA_TEST_F()
93 ExecutableRunOptions().set_device_ordinal(d).set_allocator(allocator)); in XLA_TEST_F()
Dmultiple_devices_on_host_test.cc44 execute_options.set_allocator( in CompileAndExecute()
Dlocal_client_test_base.cc153 run_options.set_allocator(GetOrCreateAllocator(local_client_->platform())); in DefaultExecutableRunOptions()
Dxla_hlo_profile_test.cc172 exec_run_options.set_allocator(backend->memory_allocator()); in ExecuteAndFetchProfile()
Dlocal_client_execute_test.cc705 DefaultExecutableRunOptions().set_allocator(&allocator)); in XLA_TEST_F()
936 run_options.set_allocator(&allocator).set_stream(stream.get()); in BM_LocalClientOverhead()
Dexhaustive_op_test.cc304 run_opts.set_allocator(client_->backend().memory_allocator()); in BuildAndRunComputation()
Ddynamic_ops_test.cc788 options.set_allocator(&allocator); in BM_DynamicSlice()
Dfusion_test.cc904 options.set_allocator(&allocator).set_stream(&stream); in BM_ParallelFusion()
Dwhile_test.cc1323 options.set_allocator(&allocator); in BM_WhileLoop()
/external/tensorflow/tensorflow/compiler/xla/
Dexecutable_run_options.cc28 ExecutableRunOptions& ExecutableRunOptions::set_allocator( in set_allocator() function in xla::ExecutableRunOptions
Dexecutable_run_options.h42 ExecutableRunOptions& set_allocator(DeviceMemoryAllocator* allocator);
/external/libxcam/xcore/
Dimage_handler.cpp38 ImageHandler::set_allocator (const SmartPtr<BufferPool> &allocator) in set_allocator() function in XCam::ImageHandler
Dimage_handler.h98 bool set_allocator (const SmartPtr<BufferPool> &allocator);
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbuffer_comparator.cc144 run_options.set_allocator(allocator_); in CompareEqualImpl()
/external/tensorflow/tensorflow/compiler/jit/kernels/
Dxla_ops.cc362 run_options.set_allocator(platform_info_.allocator()); in Compute()
536 run_options.set_allocator(platform_info_.allocator()); in Compute()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_compile_on_demand_op.cc77 run_options.set_allocator(client->backend().memory_allocator()); in Run()
/external/libxcam/modules/soft/
Dsoft_handler.cpp129 set_allocator (new SoftVideoBufAllocator); in confirm_configured()
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_execute_op.cc209 run_options.set_allocator(device_ref.backend()->memory_allocator()); in DoWork()
/external/tensorflow/tensorflow/compiler/xla/client/
Dlocal_client.cc160 run_options.set_allocator(backend_->memory_allocator()); in Run()
/external/tensorflow/tensorflow/compiler/xla/tools/
Dreplay_computation.cc288 run_options.set_allocator(&allocator); in ReplayComputation()
/external/tensorflow/tensorflow/compiler/xla/python/
Dlocal_computation_builder.cc247 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/
Dservice.cc450 options.set_allocator(backend->memory_allocator()); in ExecuteParallelAndRegisterResult()
545 options.set_allocator(backend->memory_allocator()); in ExecuteAndRegisterResult()
Dhlo_runner.cc447 run_options.set_allocator(backend().memory_allocator()); in GetServiceRunOptionsForDevice()