Searched refs:numSplits (Results 1 – 3 of 3) sorted by relevance
122 static void splitConstructor(uint32_t numSplits, uint32_t rank, RandomOperation* op) { in splitConstructor() argument128 for (uint32_t i = 0; i < numSplits; i++) { in splitConstructor()137 outDim = op->inputs[0]->dimensions[i].exactDiv(numSplits); in splitConstructor()141 for (uint32_t j = 0; j < numSplits; j++) op->outputs[j]->dimensions[i] = outDim; in splitConstructor()
2637 const size_t numSplits = splitStrs.size(); in doPackage() local2638 for (size_t i = 0; i < numSplits; i++) { in doPackage()2758 const size_t numSplits = splits.size(); in doPackage() local2759 for (size_t i = 0; i < numSplits; i++) { in doPackage()
1798 const size_t numSplits = splits.size(); in buildResources() local1799 for (size_t i = 0; i < numSplits; i++) { in buildResources()