Lines Matching refs:numOutputs
1431 const int32_t numOutputs = getScalarData<int32_t>(operands[ins[2]]); in executeOperation() local
1433 if (numOutputs != outs.size()) { in executeOperation()
1437 std::vector<Shape> outputShapes(numOutputs); in executeOperation()
1438 for (int i = 0; i < numOutputs; ++i) { in executeOperation()
1442 success = splitPrepare(input.shape(), axis, numOutputs, &outputShapes); in executeOperation()
1443 for (int i = 0; i < numOutputs; ++i) { in executeOperation()
1449 std::vector<_Float16*> outputDataPtrs(numOutputs); in executeOperation()
1450 for (int i = 0; i < numOutputs; ++i) { in executeOperation()
1458 std::vector<float*> outputDataPtrs(numOutputs); in executeOperation()
1459 for (int i = 0; i < numOutputs; ++i) { in executeOperation()
1467 std::vector<int32_t*> outputDataPtrs(numOutputs); in executeOperation()
1468 for (int i = 0; i < numOutputs; ++i) { in executeOperation()
1476 std::vector<uint8_t*> outputDataPtrs(numOutputs); in executeOperation()
1477 for (int i = 0; i < numOutputs; ++i) { in executeOperation()
1485 std::vector<int8_t*> outputDataPtrs(numOutputs); in executeOperation()
1486 for (int i = 0; i < numOutputs; ++i) { in executeOperation()