Searched refs:numSplits (Results 1 – 1 of 1) 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()