Home
last modified time | relevance | path

Searched refs:use_auto_spmd_partitioning_ (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_config.h159 use_auto_spmd_partitioning_ = use_auto_spmd_partitioning; in set_use_auto_spmd_partitioning()
172 return use_auto_spmd_partitioning_; in use_auto_spmd_partitioning()
374 bool use_auto_spmd_partitioning_ = false; variable
H A Dhlo_module.h157 use_auto_spmd_partitioning_ = use; in set_use_auto_spmd_partitioning()
161 return use_auto_spmd_partitioning_; in use_auto_spmd_partitioning()
593 bool use_auto_spmd_partitioning_ = false; variable
H A Dhlo_module.cc341 proto.set_use_auto_spmd_partitioning(use_auto_spmd_partitioning_); in ToProto()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.h96 return use_auto_spmd_partitioning_; in use_auto_spmd_partitioning()
184 bool use_auto_spmd_partitioning_ = false; variable
H A Dexecutable_build_options.cc82 use_auto_spmd_partitioning_ = use_auto_spmd_partitioning; in set_use_auto_spmd_partitioning()