Searched refs:set_auto_spmd_partitioning_mesh_shape (Results 1 – 7 of 7) sorted by relevance
104 ExecutableBuildOptions& set_auto_spmd_partitioning_mesh_shape(
87 ExecutableBuildOptions::set_auto_spmd_partitioning_mesh_shape( in set_auto_spmd_partitioning_mesh_shape() function in xla::ExecutableBuildOptions
106 config->set_auto_spmd_partitioning_mesh_shape(mesh_shape); in CreateModuleConfig()
175 void set_auto_spmd_partitioning_mesh_shape(std::vector<int64_t> mesh_shape) { in set_auto_spmd_partitioning_mesh_shape() function
544 module_config.set_auto_spmd_partitioning_mesh_shape(mesh_shape); in CreateModuleConfigFromShape()
512 config.set_auto_spmd_partitioning_mesh_shape(mesh_shape); in FromC()
726 &ExecutableBuildOptions::set_auto_spmd_partitioning_mesh_shape) in BuildXlaCompilerSubmodule()