Home
last modified time | relevance | path

Searched refs:backend_options (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbackend.cc103 BackendOptions backend_options; in CreateDefaultBackend() local
104 backend_options.set_platform(platform); in CreateDefaultBackend()
105 return CreateBackend(backend_options); in CreateDefaultBackend()
Dlocal_service.cc55 BackendOptions backend_options; in NewService() local
56 backend_options.set_platform(platform) in NewService()
61 Backend::CreateBackend(backend_options)); in NewService()
Dhlo_runner.cc38 BackendOptions backend_options; in HloRunner() local
39 backend_options.set_platform(platform); in HloRunner()
40 backend_options.set_intra_op_parallelism_threads( in HloRunner()
42 backend_ = Backend::CreateBackend(backend_options).ConsumeValueOrDie(); in HloRunner()
Dservice.cc148 BackendOptions backend_options; in NewService() local
149 backend_options.set_platform(platform); in NewService()
150 backend_options.set_allowed_devices(options.allowed_devices()); in NewService()
151 TF_ASSIGN_OR_RETURN(execute_backend, Backend::CreateBackend(backend_options)); in NewService()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dllvm_compiler_test.cc79 BackendOptions backend_options; in SetUp() local
80 backend_options.set_platform(platform); in SetUp()
82 Backend::CreateBackend(backend_options); in SetUp()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc121 xla::BackendOptions backend_options; in ConvertModule() local
122 backend_options.set_platform(platform.ValueOrDie()); in ConvertModule()
123 auto backend_or_err = xla::Backend::CreateBackend(backend_options); in ConvertModule()