Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DSqueeze.cpp36 NN_RET_CHECK(!context->isOmittedInput(kInputTensor)); in prepare()
53 if (context->isOmittedInput(kSqueezeDims)) { in prepare()
DTranspose.cpp76 NN_RET_CHECK(!context->isOmittedInput(kInputTensor)); in prepare()
87 if (context->isOmittedInput(kPermTensor)) { in prepare()
DBidirectionalSequenceRNN.cpp67 const bool hasAuxInput = !context->isOmittedInput(kAuxInputTensor); in getLinkingMode()
68 const bool hasFwAuxWeights = !context->isOmittedInput(kFwAuxWeightsTensor); in getLinkingMode()
69 const bool hasBwAuxWeights = !context->isOmittedInput(kBwAuxWeightsTensor); in getLinkingMode()
304 NN_RET_CHECK(!context->isOmittedInput(requiredInput)) in prepare()
DUnidirectionalSequenceLSTM.cpp86 NN_RET_CHECK(!context->isOmittedInput(requiredInput)) in prepare()
DQLSTM.cpp59 NN_RET_CHECK(!context->isOmittedInput(tensor)) in prepare()
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp88 bool isOmittedInput(uint32_t index) const override;
248 bool OperationExecutionContext::isOmittedInput(uint32_t index) const { in isOmittedInput() function in android::nn::__anonfe513d680111::OperationExecutionContext
258 NN_RET_CHECK(!isOmittedInput(i)) in checkNoOmittedOperand()
270 if (isOmittedInput(i)) continue; in checkNoZeroSizedInput()
/packages/modules/NeuralNetworks/common/include/
DOperationsExecutionUtils.h57 virtual bool isOmittedInput(uint32_t index) const = 0;