Home
last modified time | relevance | path

Searched refs:isOmittedOutput (Results 1 – 6 of 6) sorted by relevance

/frameworks/ml/nn/common/operations/
DSqueeze.cpp75 NN_RET_CHECK(!context->isOmittedOutput(kOutputTensor)); in prepare()
DTranspose.cpp101 NN_RET_CHECK(!context->isOmittedOutput(kOutputTensor)); in prepare()
DUnidirectionalSequenceLSTM.cpp378 NN_RET_CHECK(!context->isOmittedOutput(kOutputStateOutTensor)); in prepare()
379 NN_RET_CHECK(!context->isOmittedOutput(kCellStateOutTensor)); in prepare()
DUnidirectionalSequenceRNN.cpp189 NN_RET_CHECK(!context->isOmittedOutput(kStateOutputTensor)); in prepare()
/frameworks/ml/nn/common/include/
DOperationsUtils.h108 virtual bool isOmittedOutput(uint32_t index) const = 0;
/frameworks/ml/nn/common/
DCpuExecutor.cpp75 bool isOmittedOutput(uint32_t index) const override;
238 bool OperationExecutionContext::isOmittedOutput(uint32_t index) const { in isOmittedOutput() function in android::nn::__anonec8350e20111::OperationExecutionContext
248 NN_RET_CHECK(!isOmittedOutput(i)) << getOperationName(operation->type) << " output operand " in checkNoOmittedOperand()