Lines Matching refs:outs
740 const std::vector<uint32_t>& outs = operation.outputs; in executeOperation() local
749 auto allParametersPresent = [&operation, &operands, &ins, &outs](size_t requiredIns, in executeOperation()
785 return verify(requiredIns, ins, "in") && verify(requiredOuts, outs, "out") && in executeOperation()
801 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
819 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
883 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
945 RunTimeOperandInfo& output = operands[outs[EmbeddingLookup::kOutputTensor]]; in executeOperation()
961 RunTimeOperandInfo& output = operands[outs[HashtableLookup::kOutputTensor]]; in executeOperation()
962 RunTimeOperandInfo& hits = operands[outs[HashtableLookup::kHitsTensor]]; in executeOperation()
973 RunTimeOperandInfo& output = operands[outs[LSHProjection::kOutputTensor]]; in executeOperation()
1000 const bool output_state = (outs.size() == 5 || outs.size() == 6); in executeOperation()
1002 operands[outs[BidirectionalSequenceLSTM::kFwOutputTensor]]; in executeOperation()
1013 operands[outs[BidirectionalSequenceLSTM::kBwOutputTensor]]; in executeOperation()
1019 operands[outs[BidirectionalSequenceLSTM::kFwOutputActivationStateTensor - in executeOperation()
1022 operands[outs[BidirectionalSequenceLSTM::kFwOutputCellStateTensor - delta]]; in executeOperation()
1024 operands[outs[BidirectionalSequenceLSTM::kBwOutputActivationStateTensor - in executeOperation()
1027 operands[outs[BidirectionalSequenceLSTM::kBwOutputCellStateTensor - delta]]; in executeOperation()
1041 RunTimeOperandInfo& scratch = operands[outs[LSTMCell::kScratchBufferTensor]]; in executeOperation()
1042 RunTimeOperandInfo& outputStateOut = operands[outs[LSTMCell::kOutputStateOutTensor]]; in executeOperation()
1043 RunTimeOperandInfo& cellStateOut = operands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation()
1044 RunTimeOperandInfo& output = operands[outs[LSTMCell::kOutputTensor]]; in executeOperation()
1060 RunTimeOperandInfo& output = operands[outs[Multinomial::kOutputTensor]]; in executeOperation()
1074 RunTimeOperandInfo& hiddenStateOut = operands[outs[RNN::kHiddenStateOutTensor]]; in executeOperation()
1075 RunTimeOperandInfo& output = operands[outs[RNN::kOutputTensor]]; in executeOperation()
1085 RunTimeOperandInfo& stateOut = operands[outs[SVDF::kStateOutTensor]]; in executeOperation()
1086 RunTimeOperandInfo& output = operands[outs[SVDF::kOutputTensor]]; in executeOperation()
1104 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1176 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1252 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1291 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1306 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1344 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1360 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1368 const size_t outCount = outs.size(); in executeOperation()
1377 if (numOutputs != outs.size()) { in executeOperation()
1383 outputShapes[i] = operands[outs[i]].shape(); in executeOperation()
1388 success = success && setInfoAndAllocateIfNeeded(&(operands[outs[i]]), in executeOperation()
1395 outputDataPtrs[i] = reinterpret_cast<_Float16*>(operands[outs[i]].buffer); in executeOperation()
1404 outputDataPtrs[i] = reinterpret_cast<float*>(operands[outs[i]].buffer); in executeOperation()
1413 outputDataPtrs[i] = reinterpret_cast<int32_t*>(operands[outs[i]].buffer); in executeOperation()
1422 outputDataPtrs[i] = reinterpret_cast<uint8_t*>(operands[outs[i]].buffer); in executeOperation()
1431 outputDataPtrs[i] = reinterpret_cast<int8_t*>(operands[outs[i]].buffer); in executeOperation()
1450 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1495 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1603 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()
1620 operands[outs[QuantizedLSTMCell::kCellStateOutTensor]]; in executeOperation()
1621 RunTimeOperandInfo& output = operands[outs[QuantizedLSTMCell::kOutputTensor]]; in executeOperation()
1639 RunTimeOperandInfo& output = operands[outs[0]]; in executeOperation()