Searched defs:mesh_dim_name (Results 1 – 2 of 2) sorted by relevance
351 const std::string& mesh_dim_name, in GetMeshCoordinateIndex()362 const std::string& mesh_dim_name, in CreateConstSrcTargetPair()
1016 const std::string& mesh_dim_name = output_layout->sharding_spec(axis); in ExpandOp() local