Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_module_config.h183 auto_spmd_partitioning_mesh_ids_ = mesh_ids; in set_auto_spmd_partitioning_mesh_ids()
186 return auto_spmd_partitioning_mesh_ids_; in auto_spmd_partitioning_mesh_ids()
379 std::vector<int64_t> auto_spmd_partitioning_mesh_ids_; variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.h108 return auto_spmd_partitioning_mesh_ids_; in auto_spmd_partitioning_mesh_ids()
186 std::vector<int64_t> auto_spmd_partitioning_mesh_ids_; variable
H A Dexecutable_build_options.cc96 auto_spmd_partitioning_mesh_ids_ = mesh_ids; in set_auto_spmd_partitioning_mesh_ids()