Home
last modified time | relevance | path

Searched refs:isOmittedInput (Results 1 – 7 of 7) sorted by relevance

/frameworks/ml/nn/common/operations/
DSqueeze.cpp74 NN_RET_CHECK(!context->isOmittedInput(kInputTensor)); in prepare()
91 if (context->isOmittedInput(kSqueezeDims)) { in prepare()
DTranspose.cpp100 NN_RET_CHECK(!context->isOmittedInput(kInputTensor)); in prepare()
111 if (context->isOmittedInput(kPermTensor)) { in prepare()
DBidirectionalSequenceRNN.cpp98 const bool hasAuxInput = !context->isOmittedInput(kAuxInputTensor); in getLinkingMode()
99 const bool hasFwAuxWeights = !context->isOmittedInput(kFwAuxWeightsTensor); in getLinkingMode()
100 const bool hasBwAuxWeights = !context->isOmittedInput(kBwAuxWeightsTensor); in getLinkingMode()
364 NN_RET_CHECK(!context->isOmittedInput(requiredInput)) in prepare()
DUnidirectionalSequenceLSTM.cpp193 NN_RET_CHECK(!context->isOmittedInput(requiredInput)) in prepare()
DQLSTM.cpp174 NN_RET_CHECK(!context->isOmittedInput(tensor)) in prepare()
/frameworks/ml/nn/common/
DCpuExecutor.cpp74 bool isOmittedInput(uint32_t index) const override;
234 bool OperationExecutionContext::isOmittedInput(uint32_t index) const { in isOmittedInput() function in android::nn::__anonec8350e20111::OperationExecutionContext
244 NN_RET_CHECK(!isOmittedInput(i)) << getOperationName(operation->type) << " input operand " in checkNoOmittedOperand()
256 if (isOmittedInput(i)) continue; in checkNoZeroSizedInput()
/frameworks/ml/nn/common/include/
DOperationsUtils.h107 virtual bool isOmittedInput(uint32_t index) const = 0;