Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp1138 void quantizeOpTest(int32_t inputOperandCode, int32_t outputOperandCode) { in quantizeOpTest() argument
1141 .type = inputOperandCode, .dimensionCount = 4, .dimensions = inputDimensions}; in quantizeOpTest()
1657 void reshapeOpTest(int32_t inputOperandCode) { in reshapeOpTest() argument
1658 SCOPED_TRACE(inputOperandCode); in reshapeOpTest()
1660 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 3, inputDimensions); in reshapeOpTest()
1664 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 2, outputDimensions); in reshapeOpTest()
1677 void logSoftmaxOpTest(int32_t inputOperandCode) { in logSoftmaxOpTest() argument
1679 ANeuralNetworksOperandType input = {.type = inputOperandCode, in logSoftmaxOpTest()
1684 ANeuralNetworksOperandType beta = {.type = (inputOperandCode == ANEURALNETWORKS_TENSOR_FLOAT32) in logSoftmaxOpTest()
1697 ANeuralNetworksOperandType output = {.type = inputOperandCode, in logSoftmaxOpTest()
[all …]