Searched defs:set_auto_spmd_partitioning_mesh_shape (Results 1 – 2 of 2) sorted by relevance
87 ExecutableBuildOptions::set_auto_spmd_partitioning_mesh_shape( in set_auto_spmd_partitioning_mesh_shape() function in xla::ExecutableBuildOptions
175 void set_auto_spmd_partitioning_mesh_shape(std::vector<int64_t> mesh_shape) { in set_auto_spmd_partitioning_mesh_shape() function