Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/test/
DTestMemory.cpp59 WrapperOperandType matrixType(WrapperType::TENSOR_FLOAT32, {3, 4}); in TEST_F() local
62 auto a = model.addOperand(&matrixType); in TEST_F()
63 auto b = model.addOperand(&matrixType); in TEST_F()
64 auto c = model.addOperand(&matrixType); in TEST_F()
65 auto d = model.addOperand(&matrixType); in TEST_F()
66 auto e = model.addOperand(&matrixType); in TEST_F()
117 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()
[all …]
DTestTrivialModel.cpp61 OperandType matrixType(Type::TENSOR_FLOAT32, {3, 4}); in CreateAddTwoTensorModel() local
63 auto a = model->addOperand(&matrixType); in CreateAddTwoTensorModel()
64 auto b = model->addOperand(&matrixType); in CreateAddTwoTensorModel()
65 auto c = model->addOperand(&matrixType); in CreateAddTwoTensorModel()
76 OperandType matrixType(Type::TENSOR_FLOAT32, {3, 4}); in CreateAddThreeTensorModel() local
78 auto a = model->addOperand(&matrixType); in CreateAddThreeTensorModel()
79 auto b = model->addOperand(&matrixType); in CreateAddThreeTensorModel()
80 auto c = model->addOperand(&matrixType); in CreateAddThreeTensorModel()
81 auto d = model->addOperand(&matrixType); in CreateAddThreeTensorModel()
82 auto e = model->addOperand(&matrixType); in CreateAddThreeTensorModel()
[all …]
DTestMemoryInternal.cpp118 WrapperOperandType matrixType(WrapperType::TENSOR_FLOAT32, {3, 4}); in TEST_F() local
121 auto a = model.addOperand(&matrixType); in TEST_F()
122 auto b = model.addOperand(&matrixType); in TEST_F()
123 auto c = model.addOperand(&matrixType); in TEST_F()
124 auto d = model.addOperand(&matrixType); in TEST_F()
125 auto e = model.addOperand(&matrixType); in TEST_F()
DTestCompilationCaching.cpp270 test_wrapper::OperandType matrixType(Type::TENSOR_FLOAT32, {2, 2}); in CreateBroadcastAddModel() local
274 auto a = model->addOperand(&matrixType); in CreateBroadcastAddModel()
276 auto c = model->addOperand(&matrixType); in CreateBroadcastAddModel()