Searched refs:ExtensionOperandType (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/runtime/test/fibonacci_extension/ |
| D | FibonacciExtensionTest.cpp | 40 using ::android::nn::test_wrapper::ExtensionOperandType;
123 void addNopOperation(ExtensionModel* model, ExtensionOperandType inputType, uint32_t input, in addNopOperation()
136 ExtensionOperandType scalarType(Type::INT32, {}); in addNopOperation()
144 void createModel(ExtensionModel* model, ExtensionOperandType inputType, in createModel()
145 ExtensionOperandType outputType, bool addNopOperations) { in createModel()
172 ExtensionOperandType inputType(static_cast<Type>(mModel.getExtensionOperandType( in TEST_F()
175 ExtensionOperandType outputType( in TEST_F()
212 ExtensionOperandType inputType(Type::TENSOR_FLOAT32, {1}); in TEST_F()
213 ExtensionOperandType outputType(Type::TENSOR_FLOAT32, {N}); in TEST_F()
241 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 | 45 using wrapper::ExtensionOperandType;
|