Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp99 const bool hasAuxInputs = !context->isOmittedInput(kAuxInputTensor); in executeTyped()
287 NN_RET_CHECK(!context->isOmittedInput(requiredInput)) in prepare()
305 const bool auxInputsAllOrNone = (context->isOmittedInput(kAuxInputTensor) && in prepare()
306 context->isOmittedInput(kFwAuxWeightsTensor) && in prepare()
307 context->isOmittedInput(kBwAuxWeightsTensor)) || in prepare()
308 (!context->isOmittedInput(kAuxInputTensor) && in prepare()
309 !context->isOmittedInput(kFwAuxWeightsTensor) && in prepare()
310 !context->isOmittedInput(kBwAuxWeightsTensor)); in prepare()
312 const bool hasAuxInputs = !context->isOmittedInput(kAuxInputTensor); in prepare()
DTranspose.cpp87 NN_RET_CHECK(!context->isOmittedInput(kInputTensor)); in prepare()
98 if (context->isOmittedInput(kPermTensor)) { in prepare()
DUnidirectionalSequenceLSTM.cpp180 NN_RET_CHECK(!context->isOmittedInput(requiredInput)) in prepare()
/frameworks/ml/nn/common/include/
DOperationsUtils.h103 virtual bool isOmittedInput(uint32_t index) const = 0;
/frameworks/ml/nn/common/
DCpuExecutor.cpp63 bool isOmittedInput(uint32_t index) const override;
212 bool OperationExecutionContext::isOmittedInput(uint32_t index) const { in isOmittedInput() function in android::nn::__anon6e12a00b0111::OperationExecutionContext
222 NN_RET_CHECK(!isOmittedInput(i)) << getOperationName(operation->type) << " input operand " in checkNoOmittedOperand()
234 if (isOmittedInput(i)) continue; in checkNoZeroSizedInput()