Home
last modified time | relevance | path

Searched refs:use_spmd_partitioning_ (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_config.h154 use_spmd_partitioning_ = use_spmd_partitioning; in set_use_spmd_partitioning()
156 bool use_spmd_partitioning() const { return use_spmd_partitioning_; } in use_spmd_partitioning()
371 bool use_spmd_partitioning_ = false; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.h91 bool use_spmd_partitioning() const { return use_spmd_partitioning_; } in use_spmd_partitioning()
183 bool use_spmd_partitioning_ = false; variable
H A Dexecutable_build_options.cc76 use_spmd_partitioning_ = use_spmd_partitioning; in set_use_spmd_partitioning()