Home
last modified time | relevance | path

Searched refs:set_use_auto_spmd_partitioning (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.cc80 ExecutableBuildOptions& ExecutableBuildOptions::set_use_auto_spmd_partitioning( in set_use_auto_spmd_partitioning() function in xla::ExecutableBuildOptions
118 output.set_use_auto_spmd_partitioning(use_auto_spmd_partitioning()); in ToProto()
147 output.set_use_auto_spmd_partitioning(input.use_auto_spmd_partitioning()); in ExecutableBuildOptionsFromProto()
199 execution_options.set_use_auto_spmd_partitioning( in CreateExecutionOptions()
H A Dexecutable_build_options.h98 ExecutableBuildOptions& set_use_auto_spmd_partitioning(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module.cc341 proto.set_use_auto_spmd_partitioning(use_auto_spmd_partitioning_); in ToProto()
507 module->set_use_auto_spmd_partitioning(proto.use_auto_spmd_partitioning()); in CreateFromProto()
538 module_config.set_use_auto_spmd_partitioning( in CreateModuleConfigFromShape()
H A Dcompile_only_service.cc92 execution_options.set_use_auto_spmd_partitioning( in CompileAheadOfTime()
H A Dhlo_module_util.cc100 config->set_use_auto_spmd_partitioning( in CreateModuleConfig()
H A Dhlo_module_config.h158 void set_use_auto_spmd_partitioning(bool use_auto_spmd_partitioning) { in set_use_auto_spmd_partitioning() function
H A Dhlo_module.h156 void set_use_auto_spmd_partitioning(bool use) { in set_use_auto_spmd_partitioning() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dc_api_conversions.cc507 config.set_use_auto_spmd_partitioning(c_config.use_auto_spmd_partitioning); in FromC()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dxla_compiler.cc722 &ExecutableBuildOptions::set_use_auto_spmd_partitioning) in BuildXlaCompilerSubmodule()