Home
last modified time | relevance | path

Searched defs:execution_options (Results 1 – 21 of 21) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dservice.cc263 const ExecutionOptions* execution_options, in CreateModuleConfig()
280 const ExecutionOptions& execution_options, in CreateModuleConfig()
557 const ExecutionOptions& execution_options, int64_t requests_size, in GetExecutors()
582 const ExecutionOptions& execution_options, in GetArguments()
623 const ExecutionOptions& execution_options = in ExecuteGraphParallel() local
H A Dhlo_module.cc526 const ExecutionOptions* execution_options) { in CreateModuleConfigFromShape()
586 const ExecutionOptions* execution_options) { in CreateModuleConfigFromProto()
H A Dcompile_only_service.cc69 ExecutionOptions execution_options; in CompileAheadOfTime() local
H A Dhlo_module_util.cc47 const ExecutionOptions* execution_options, int default_num_replicas, in CreateModuleConfig()
H A Dlocal_service.cc144 ExecutionOptions execution_options = in GetHloModuleConfig() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dclient_test.cc49 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F() local
79 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F() local
H A Dprng_test.cc214 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F() local
307 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F() local
H A Dclient_library_test_base.cc114 ExecutionOptions execution_options = execution_options_; in ExecuteAndTransfer() local
134 ExecutionOptions execution_options = execution_options_; in ExecuteAndTransferReference() local
H A Dreshape_test.cc650 ExecutionOptions execution_options = execution_options_; in XLA_TEST_P() local
806 ExecutionOptions execution_options = execution_options_; in XLA_TEST_P() local
H A Dgather_operation_test.cc774 xla::ExecutionOptions execution_options = CreateDefaultExecutionOptions(); in XLA_TEST_F() local
H A Dwhile_test.cc927 ExecutionOptions execution_options = execution_options_; in XLA_TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dclient.cc167 const ExecutionOptions* execution_options, in ExecuteAndTransfer()
218 const ExecutionOptions* execution_options) { in Compile()
278 const ExecutionOptions* execution_options, in Execute()
H A Dcompile_only_client.cc30 const ExecutionOptions* execution_options) { in CreateModuleConfig()
H A Dclient.h102 ExecutionOptions execution_options; member
H A Dexecutable_build_options.cc182 ExecutionOptions execution_options = CreateDefaultExecutionOptions(); in CreateExecutionOptions() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dexecution_options_util.cc21 ExecutionOptions execution_options; in CreateDefaultExecutionOptions() local
H A Dxla.proto604 ExecutionOptions execution_options = 2; field
632 ExecutionOptions execution_options = 3; field
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dtesting.cc69 auto execution_options = CreateDefaultExecutionOptions(); in MakeFakeDataViaDeviceOrDie() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtfrt_cpu_pjrt_client.cc250 const ExecutionOptions& execution_options) { in JitCompile()
387 ExecutionOptions execution_options = in Compile() local
/aosp_15_r20/tools/asuite/atest/bazel/runner/src/main/protobuf/
H A Dfailure_details.proto103 ExecutionOptions execution_options = 121; field
/aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/
H A Dmaster_session.cc1278 GraphExecutionStateOptions execution_options; in Create() local