Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/runtime/test/
H A DTestMemory.cpp60 WrapperOperandType matrixType(WrapperType::TENSOR_FLOAT32, {3, 4}); in TEST_F() local
63 auto a = model.addOperand(&matrixType); in TEST_F()
64 auto b = model.addOperand(&matrixType); in TEST_F()
65 auto c = model.addOperand(&matrixType); in TEST_F()
66 auto d = model.addOperand(&matrixType); in TEST_F()
67 auto e = model.addOperand(&matrixType); in TEST_F()
126 auto a = model.addOperand(&matrixType); in TEST_F()
127 auto b = model.addOperand(&matrixType); in TEST_F()
128 auto c = model.addOperand(&matrixType); in TEST_F()
129 auto d = model.addOperand(&matrixType); in TEST_F()
[all …]
H A DTestTrivialModel.cpp65 OperandType matrixType(Type::TENSOR_FLOAT32, {3, 4}); in CreateAddTwoTensorModel() local
67 auto a = model->addOperand(&matrixType); in CreateAddTwoTensorModel()
68 auto b = model->addOperand(&matrixType); in CreateAddTwoTensorModel()
69 auto c = model->addOperand(&matrixType); in CreateAddTwoTensorModel()
80 OperandType matrixType(Type::TENSOR_FLOAT32, {3, 4}); in CreateAddThreeTensorModel() local
82 auto a = model->addOperand(&matrixType); in CreateAddThreeTensorModel()
83 auto b = model->addOperand(&matrixType); in CreateAddThreeTensorModel()
84 auto c = model->addOperand(&matrixType); in CreateAddThreeTensorModel()
85 auto d = model->addOperand(&matrixType); in CreateAddThreeTensorModel()
86 auto e = model->addOperand(&matrixType); in CreateAddThreeTensorModel()
[all …]
H A DTestMemoryInternal.cpp117 WrapperOperandType matrixType(WrapperType::TENSOR_FLOAT32, {3, 4}); in TEST_F() local
120 auto a = model.addOperand(&matrixType); in TEST_F()
121 auto b = model.addOperand(&matrixType); in TEST_F()
122 auto c = model.addOperand(&matrixType); in TEST_F()
123 auto d = model.addOperand(&matrixType); in TEST_F()
124 auto e = model.addOperand(&matrixType); in TEST_F()
H A DTestCompilationCaching.cpp280 test_wrapper::OperandType matrixType(Type::TENSOR_FLOAT32, {2, 2}); in CreateBroadcastAddModel() local
284 auto a = model->addOperand(&matrixType); in CreateBroadcastAddModel()
286 auto c = model->addOperand(&matrixType); in CreateBroadcastAddModel()