Searched refs:hasAuxInput (Results 1 – 4 of 4) sorted by relevance
67 const bool hasAuxInput = !context->isOmittedInput(kAuxInputTensor); in getLinkingMode() local77 if (!hasAuxInput && !hasFwAuxWeights && !hasBwAuxWeights) { in getLinkingMode()79 } else if (hasAuxInput && !hasFwAuxWeights && !hasBwAuxWeights) { in getLinkingMode()81 } else if (hasAuxInput && hasFwAuxWeights && hasBwAuxWeights) { in getLinkingMode()122 const bool hasAuxInput = (linkingMode == LinkingMode::CROSS_LINKING || in executeTyped() local152 if (hasAuxInput) { in executeTyped()162 if (hasAuxInput) { in executeTyped()168 if (hasAuxInput) { in executeTyped()179 if (hasAuxInput) { in executeTyped()192 if (hasAuxInput) { in executeTyped()[all …]
153 bool hasAuxInput = (auxInputData != nullptr); in RNNStep() local154 if (hasAuxInput) { in RNNStep()165 if (hasAuxInput) { in RNNStep()174 if (hasAuxInput) { in RNNStep()191 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()
57 bool getLinkingMode(bool hasAuxInput, bool hasAuxWeights, LinkingMode* linkingMode) { in getLinkingMode() argument64 if (!hasAuxInput && !hasAuxWeights) { in getLinkingMode()66 } else if (hasAuxInput && !hasAuxWeights) { in getLinkingMode()68 } else if (hasAuxInput && hasAuxWeights) { in getLinkingMode()