Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp1139 void quantizeOpTest(int32_t inputOperandCode, int32_t outputOperandCode) { in quantizeOpTest() argument
1142 .type = inputOperandCode, .dimensionCount = 4, .dimensions = inputDimensions}; in quantizeOpTest()
1658 void reshapeOpTest(int32_t inputOperandCode) { in reshapeOpTest() argument
1659 SCOPED_TRACE(inputOperandCode); in reshapeOpTest()
1661 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 3, inputDimensions); in reshapeOpTest()
1665 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 2, outputDimensions); in reshapeOpTest()
1678 void logSoftmaxOpTest(int32_t inputOperandCode) { in logSoftmaxOpTest() argument
1680 ANeuralNetworksOperandType input = {.type = inputOperandCode, in logSoftmaxOpTest()
1685 ANeuralNetworksOperandType beta = {.type = (inputOperandCode == ANEURALNETWORKS_TENSOR_FLOAT32) in logSoftmaxOpTest()
1698 ANeuralNetworksOperandType output = {.type = inputOperandCode, in logSoftmaxOpTest()
[all …]