Searched defs:partialAxes (Results 1 – 2 of 2) sorted by relevance
293 ArrayRef<MeshAxis> partialAxes = shardAttr.getPartialAxes(); defaultGetShardingOption() local 398 SmallVector<MeshAxis> partialAxes; getShardingAttribute() local [all...]
340 verify(function_ref<InFlightDiagnostic ()> emitError,FlatSymbolRefAttr,ArrayRef<MeshAxesAttr> splitAxes,ArrayRef<MeshAxis> partialAxes,ReductionKind) verify() argument