Searched refs:auto_spmd_partitioning_mesh_shape (Results 1 – 12 of 12) sorted by relevance
462 CreateVector(config.auto_spmd_partitioning_mesh_shape(), in ToC()463 &hlo_config.auto_spmd_partitioning_mesh_shape); in ToC()509 MakeSpan(c_config.auto_spmd_partitioning_mesh_shape); in FromC()
310 Int64List auto_spmd_partitioning_mesh_shape; member
101 std::vector<int64_t> auto_spmd_partitioning_mesh_shape() const { in auto_spmd_partitioning_mesh_shape() function
201 for (auto t : build_options.auto_spmd_partitioning_mesh_shape()) { in CreateExecutionOptions()
94 for (auto t : options.auto_spmd_partitioning_mesh_shape()) { in CompileAheadOfTime()
103 for (auto t : execution_options->auto_spmd_partitioning_mesh_shape()) { in CreateModuleConfig()
178 std::vector<int64_t> auto_spmd_partitioning_mesh_shape() const { in auto_spmd_partitioning_mesh_shape() function
94 virtual std::vector<int64_t> auto_spmd_partitioning_mesh_shape() const { in auto_spmd_partitioning_mesh_shape() function
541 for (auto t : execution_options->auto_spmd_partitioning_mesh_shape()) { in CreateModuleConfigFromShape()
253 auto_spmd_partitioning_mesh_shape: List[int]
725 &ExecutableBuildOptions::auto_spmd_partitioning_mesh_shape, in BuildXlaCompilerSubmodule()
498 repeated int64 auto_spmd_partitioning_mesh_shape = 16; field