Searched refs:ExtensionOperandType (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/runtime/test/fibonacci_extension/ |
| D | FibonacciExtensionTest.cpp | 38 using ::android::nn::test_wrapper::ExtensionOperandType;
121 void addNopOperation(ExtensionModel* model, ExtensionOperandType inputType, uint32_t input, in addNopOperation()
134 ExtensionOperandType scalarType(Type::INT32, {}); in addNopOperation()
142 void createModel(ExtensionModel* model, ExtensionOperandType inputType, in createModel()
143 ExtensionOperandType outputType, bool addNopOperations) { in createModel()
170 ExtensionOperandType inputType( in TEST_F()
174 ExtensionOperandType outputType( in TEST_F()
211 ExtensionOperandType inputType(Type::TENSOR_FLOAT32, {1}); in TEST_F()
212 ExtensionOperandType outputType(Type::TENSOR_FLOAT32, {N}); in TEST_F()
240 ExtensionOperandType inputType(Type::TENSOR_INT32, {1}); // Unsupported type. in TEST_F()
[all …]
|
| /frameworks/ml/nn/runtime/include/ |
| D | NeuralNetworksWrapperExtensions.h | 143 using ExtensionOperandType = extension_wrapper::OperandType; variable
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestNeuralNetworksWrapper.h | 40 using wrapper::ExtensionOperandType;
|