Lines Matching defs:meshAxis

191     auto [tensorAxis, meshAxis] = detectRes.value();
193                                      tensorAxis, meshAxis);
290     auto [tensorAxis, meshAxis] = detectRes.value();
293                                        tensorAxis, meshAxis);
360   auto meshAxis = sourceSplitAxes.back();
367   targetSplitAxes.push_back(meshAxis);
394                               int64_t targetTensorAxis, MeshAxis meshAxis) {
401       sourceShard.getType(), mesh.getShape()[meshAxis], sourceTensorAxis,
406       mesh.getSymName(), SmallVector<MeshAxis>({meshAxis}), sourceShard,
423     auto [sourceTensorAxis, targetTensorAxis, meshAxis] = detectRes.value();
426         sourceTensorAxis, targetTensorAxis, meshAxis);