Lines Matching refs:outs
741 const std::vector<uint32_t>& outs = operation.outputs; in executeOperation() local
750 auto allParametersPresent = [&operation, &operands, &ins, &outs](size_t requiredIns, in executeOperation()
786 return verify(requiredIns, ins, "in") && verify(requiredOuts, outs, "out") && in executeOperation()
802 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
820 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
884 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
946 RunTimeOperandInfo& output = operands[outs[EmbeddingLookup::kOutputTensor]]; in executeOperation()
962 RunTimeOperandInfo& output = operands[outs[HashtableLookup::kOutputTensor]]; in executeOperation()
963 RunTimeOperandInfo& hits = operands[outs[HashtableLookup::kHitsTensor]]; in executeOperation()
974 RunTimeOperandInfo& output = operands[outs[LSHProjection::kOutputTensor]]; in executeOperation()
1001 const bool output_state = (outs.size() == 5 || outs.size() == 6); in executeOperation()
1003 operands[outs[BidirectionalSequenceLSTM::kFwOutputTensor]]; in executeOperation()
1014 operands[outs[BidirectionalSequenceLSTM::kBwOutputTensor]]; in executeOperation()
1020 operands[outs[BidirectionalSequenceLSTM::kFwOutputActivationStateTensor - in executeOperation()
1023 operands[outs[BidirectionalSequenceLSTM::kFwOutputCellStateTensor - delta]]; in executeOperation()
1025 operands[outs[BidirectionalSequenceLSTM::kBwOutputActivationStateTensor - in executeOperation()
1028 operands[outs[BidirectionalSequenceLSTM::kBwOutputCellStateTensor - delta]]; in executeOperation()
1042 RunTimeOperandInfo& scratch = operands[outs[LSTMCell::kScratchBufferTensor]]; in executeOperation()
1043 RunTimeOperandInfo& outputStateOut = operands[outs[LSTMCell::kOutputStateOutTensor]]; in executeOperation()
1044 RunTimeOperandInfo& cellStateOut = operands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation()
1045 RunTimeOperandInfo& output = operands[outs[LSTMCell::kOutputTensor]]; in executeOperation()
1061 RunTimeOperandInfo& output = operands[outs[Multinomial::kOutputTensor]]; in executeOperation()
1075 RunTimeOperandInfo& hiddenStateOut = operands[outs[RNN::kHiddenStateOutTensor]]; in executeOperation()
1076 RunTimeOperandInfo& output = operands[outs[RNN::kOutputTensor]]; in executeOperation()
1086 RunTimeOperandInfo& stateOut = operands[outs[SVDF::kStateOutTensor]]; in executeOperation()
1087 RunTimeOperandInfo& output = operands[outs[SVDF::kOutputTensor]]; in executeOperation()
1105 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1177 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1253 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1292 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1307 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1345 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1361 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1369 const size_t outCount = outs.size(); in executeOperation()
1378 if (static_cast<size_t>(numOutputs) != outs.size()) { in executeOperation()
1384 outputShapes[i] = operands[outs[i]].shape(); in executeOperation()
1389 success = success && setInfoAndAllocateIfNeeded(&(operands[outs[i]]), in executeOperation()
1396 outputDataPtrs[i] = reinterpret_cast<_Float16*>(operands[outs[i]].buffer); in executeOperation()
1405 outputDataPtrs[i] = reinterpret_cast<float*>(operands[outs[i]].buffer); in executeOperation()
1414 outputDataPtrs[i] = reinterpret_cast<int32_t*>(operands[outs[i]].buffer); in executeOperation()
1423 outputDataPtrs[i] = reinterpret_cast<uint8_t*>(operands[outs[i]].buffer); in executeOperation()
1432 outputDataPtrs[i] = reinterpret_cast<int8_t*>(operands[outs[i]].buffer); in executeOperation()
1451 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1496 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1604 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1621 operands[outs[QuantizedLSTMCell::kCellStateOutTensor]]; in executeOperation()
1622 RunTimeOperandInfo& output = operands[outs[QuantizedLSTMCell::kOutputTensor]]; in executeOperation()
1640 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()