Lines Matching refs:mOpCode

336         : mOpCode(opCode), mValidInputs(), mValidOutputs(), mInputRankMutators(inputRankMutators) {
391 model, mOpCode, static_cast<uint32_t>(inputIds.size()), inputIds.data(), in addOperation()
423 if (mOpCode == ANEURALNETWORKS_LSH_PROJECTION && i == 1) { in testMutatingInputOperandCode()
427 if (mOpCode == ANEURALNETWORKS_RANK) { in testMutatingInputOperandCode()
435 if ((mOpCode == ANEURALNETWORKS_TRANSPOSE_CONV_2D || in testMutatingInputOperandCode()
436 mOpCode == ANEURALNETWORKS_DEPTHWISE_CONV_2D) && in testMutatingInputOperandCode()
447 if (mOpCode == ANEURALNETWORKS_CAST && i == 0) { in testMutatingInputOperandCode()
456 if (mOpCode == ANEURALNETWORKS_RANDOM_MULTINOMIAL && i == 0) { in testMutatingInputOperandCode()
465 if (mOpCode == ANEURALNETWORKS_DEQUANTIZE && i == 0) { in testMutatingInputOperandCode()
473 if (mOpCode == ANEURALNETWORKS_AXIS_ALIGNED_BBOX_TRANSFORM && i == 1) { in testMutatingInputOperandCode()
486 if (mOpCode == ANEURALNETWORKS_CONV_2D && i == 7 && mValidInputs.size() == 10) { in testMutatingInputOperandCode()
496 if (mOpCode == ANEURALNETWORKS_QUANTIZE && i == 0 && in testMutatingInputOperandCode()
503 if ((mOpCode == ANEURALNETWORKS_ARGMIN || mOpCode == ANEURALNETWORKS_ARGMAX) && in testMutatingInputOperandCode()
516 if (mOpCode == ANEURALNETWORKS_DEPTHWISE_CONV_2D && i == 8 && in testMutatingInputOperandCode()
545 if (mOpCode == ANEURALNETWORKS_LSH_PROJECTION && i == 1) { in testMutatingOutputOperandCode()
555 if (mOpCode == ANEURALNETWORKS_DEQUANTIZE && in testMutatingOutputOperandCode()
563 if (mOpCode == ANEURALNETWORKS_QUANTIZE && i == 0 && in testMutatingOutputOperandCode()
570 if (mOpCode == ANEURALNETWORKS_CAST && i == 0 && in testMutatingOutputOperandCode()
592 if (mOpCode == ANEURALNETWORKS_LSTM) { in testMutatingInputOperandCounts()
608 if (mOpCode == ANEURALNETWORKS_SPLIT) { in testMutatingOutputOperandCounts()
615 if (mOpCode == ANEURALNETWORKS_UNIDIRECTIONAL_SEQUENCE_RNN && i < 1) { in testMutatingOutputOperandCounts()
618 if (mOpCode == ANEURALNETWORKS_UNIDIRECTIONAL_SEQUENCE_LSTM && i < 3) { in testMutatingOutputOperandCounts()
621 if (mOpCode == ANEURALNETWORKS_BIDIRECTIONAL_SEQUENCE_RNN && i < 3) { in testMutatingOutputOperandCounts()
624 if (mOpCode == ANEURALNETWORKS_BIDIRECTIONAL_SEQUENCE_LSTM && i < 5) { in testMutatingOutputOperandCounts()
654 ANeuralNetworksOperationType mOpCode; member in __anon37b312a30111::OperationTestBase