Searched defs:tensorAxis (Results 1 – 2 of 2) sorted by relevance
155 for (size_t tensorAxis = 0; tensorAxis < targetSharding.getSplitAxes().size(); in detectSplitLastAxisInResharding() local 205 for (size_t tensorAxis = 0; tensorAxis < sourceSharding.getSplitAxes().size(); detectUnsplitLastAxisInResharding() local [all...]
649 verifyScatterOrSliceOperandAndResultShape(Value operand,Value result,int64_t tensorAxis,ArrayRef<MeshAxis> meshAxes,ArrayRef<int64_t> meshShape) verifyScatterOrSliceOperandAndResultShape() argument [all...]