Home
last modified time | relevance | path

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

/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/
DTestValidateOperations.cpp1149 void quantizeOpTest(int32_t inputOperandCode, int32_t outputOperandCode) { in quantizeOpTest() argument
1152 .type = inputOperandCode, .dimensionCount = 4, .dimensions = inputDimensions}; in quantizeOpTest()
1676 void reshapeOpTest(int32_t inputOperandCode) { in reshapeOpTest() argument
1677 SCOPED_TRACE(inputOperandCode); in reshapeOpTest()
1679 ANeuralNetworksOperandType input = getOpType(inputOperandCode, 3, inputDimensions); in reshapeOpTest()
1683 ANeuralNetworksOperandType output = getOpType(inputOperandCode, 2, outputDimensions); in reshapeOpTest()
1697 void logSoftmaxOpTest(int32_t inputOperandCode) { in logSoftmaxOpTest() argument
1699 ANeuralNetworksOperandType input = {.type = inputOperandCode, in logSoftmaxOpTest()
1704 ANeuralNetworksOperandType beta = {.type = (inputOperandCode == ANEURALNETWORKS_TENSOR_FLOAT32) in logSoftmaxOpTest()
1717 ANeuralNetworksOperandType output = {.type = inputOperandCode, in logSoftmaxOpTest()
[all …]