Searched defs:mesh_shape (Results 1 – 3 of 3) sorted by relevance
12 repeated int32 mesh_shape = 1; field
52 def __init__(self, serialized=None, mesh_shape=None, device_coordinates=None): argument150 def mesh_shape(self): member in Topology
114 std::string TopologyWithMeshShape(llvm::ArrayRef<int> mesh_shape) { in TopologyWithMeshShape()120 std::string TopologyWithMeshShapeAndTasks(llvm::ArrayRef<int> mesh_shape, in TopologyWithMeshShapeAndTasks()