Searched refs:ExtensionOperandType (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/runtime/test/fibonacci_extension/ |
| D | FibonacciExtensionTest.cpp | 39 using ::android::nn::test_wrapper::ExtensionOperandType;
129 void addNopOperation(ExtensionModel* model, ExtensionOperandType inputType, uint32_t input, in addNopOperation()
142 ExtensionOperandType scalarType(Type::INT32, {}); in addNopOperation()
150 void createModel(ExtensionModel* model, ExtensionOperandType inputType, in createModel()
151 ExtensionOperandType outputType, bool addNopOperations) { in createModel()
178 ExtensionOperandType inputType(static_cast<Type>(mModel.getExtensionOperandType( in TEST_F()
181 ExtensionOperandType outputType( in TEST_F()
218 ExtensionOperandType inputType(Type::TENSOR_FLOAT32, {1}); in TEST_F()
219 ExtensionOperandType outputType(Type::TENSOR_FLOAT32, {N}); in TEST_F()
247 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;
|