Searched refs:hasAuxInput (Results 1 – 4 of 4) sorted by relevance
98 const bool hasAuxInput = !context->isOmittedInput(kAuxInputTensor); in getLinkingMode() local108 if (!hasAuxInput && !hasFwAuxWeights && !hasBwAuxWeights) { in getLinkingMode()110 } else if (hasAuxInput && !hasFwAuxWeights && !hasBwAuxWeights) { in getLinkingMode()112 } else if (hasAuxInput && hasFwAuxWeights && hasBwAuxWeights) { in getLinkingMode()153 const bool hasAuxInput = (linkingMode == LinkingMode::CROSS_LINKING || in executeTyped() local183 if (hasAuxInput) { in executeTyped()193 if (hasAuxInput) { in executeTyped()199 if (hasAuxInput) { in executeTyped()210 if (hasAuxInput) { in executeTyped()223 if (hasAuxInput) { in executeTyped()[all …]
157 bool hasAuxInput = (auxInputData != nullptr); in RNNStep() local158 if (hasAuxInput) { in RNNStep()169 if (hasAuxInput) { in RNNStep()178 if (hasAuxInput) { in RNNStep()195 if (hasAuxInput) { in RNNStep()
446 const bool hasAuxInput = (aux_input_buffer != nullptr); in LSTMEvalFloat32() local454 if (hasAuxInput) { in LSTMEvalFloat32()466 hasAuxInput ? (timeMajor ? aux_input_buffer : transposedAuxInput.data()) : nullptr; in LSTMEvalFloat32()476 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat32()502 if (hasAuxInput) { in LSTMEvalFloat32()681 const bool hasAuxInput = (aux_input_buffer != nullptr); in LSTMEvalFloat16() local690 if (hasAuxInput) { in LSTMEvalFloat16()702 hasAuxInput ? (timeMajor ? aux_input_float32.data() : transposedAuxInput.data()) in LSTMEvalFloat16()714 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat16()744 if (hasAuxInput) { in LSTMEvalFloat16()
58 bool getLinkingMode(bool hasAuxInput, bool hasAuxWeights, LinkingMode* linkingMode) { in getLinkingMode() argument65 if (!hasAuxInput && !hasAuxWeights) { in getLinkingMode()67 } else if (hasAuxInput && !hasAuxWeights) { in getLinkingMode()69 } else if (hasAuxInput && hasAuxWeights) { in getLinkingMode()