Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/client/
Dexecutable_build_options.h96 return use_auto_spmd_partitioning_; in use_auto_spmd_partitioning()
184 bool use_auto_spmd_partitioning_ = false; variable
Dexecutable_build_options.cc82 use_auto_spmd_partitioning_ = use_auto_spmd_partitioning; in set_use_auto_spmd_partitioning()
/external/tensorflow/tensorflow/compiler/xla/service/
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
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
Dhlo_module.cc341 proto.set_use_auto_spmd_partitioning(use_auto_spmd_partitioning_); in ToProto()