Searched defs:mesh_config (Results 1 – 7 of 7) sorted by relevance
148 : mesh_config_(mesh_config), in mesh_config_() argument166 const dtensor::Mesh& mesh_config() const { return mesh_config_; } in mesh_config() function
2032 auto mesh_config = mesh_config_or_status.ValueOrDie(); in AddMesh() local
40 MeshProto mesh_config = 2; field
82 const auto mesh_config = status_or_mesh.ValueOrDie(); in WrapDeviceCluster() local
56 void AddMetadataToTPUCluster(const Mesh& mesh_config, in AddMetadataToTPUCluster()
76 mlir::LogicalResult UpdateTPUCompileMetadata(const Mesh& mesh_config, in UpdateTPUCompileMetadata()
59 const u8 *mesh_config; member