Home
last modified time | relevance | path

Searched refs:WrapperType (Results 1 – 7 of 7) sorted by relevance

/frameworks/ml/nn/runtime/test/
DTestMemoryInternal.cpp39 using WrapperType = ::android::nn::test_wrapper::Type; typedef
118 WrapperOperandType matrixType(WrapperType::TENSOR_FLOAT32, {3, 4}); in TEST_F()
119 WrapperOperandType scalarType(WrapperType::INT32, {}); in TEST_F()
237 WrapperOperandType type3(WrapperType::INT32, {}); in TEST_F()
238 WrapperOperandType type2(WrapperType::TENSOR_INT32, {128}, 0.25, 0); in TEST_F()
239 WrapperOperandType type0(WrapperType::TENSOR_QUANT8_ASYMM, {12, 104, 104, 128}, 0.5, 0); in TEST_F()
240 WrapperOperandType type4(WrapperType::TENSOR_QUANT8_ASYMM, {12, 92, 92, 128}, 1.0, 0); in TEST_F()
241 WrapperOperandType type1(WrapperType::TENSOR_QUANT8_ASYMM, {128, 13, 13, 128}, 0.5, 0); in TEST_F()
DTestMemory.cpp32 using WrapperType = ::android::nn::test_wrapper::Type; typedef
60 WrapperOperandType matrixType(WrapperType::TENSOR_FLOAT32, {3, 4}); in TEST_F()
61 WrapperOperandType scalarType(WrapperType::INT32, {}); in TEST_F()
118 WrapperOperandType matrixType(WrapperType::TENSOR_FLOAT32, {3, 4}); in TEST_F()
119 WrapperOperandType scalarType(WrapperType::INT32, {}); in TEST_F()
DTestCompliance.cpp31 using WrapperType = test_wrapper::Type; typedef
62 static const WrapperOperandType kTypeTensorFloat(WrapperType::TENSOR_FLOAT32, {1});
63 static const WrapperOperandType kTypeTensorFloatRank0(WrapperType::TENSOR_FLOAT32, {});
64 static const WrapperOperandType kTypeInt32(WrapperType::INT32, {});
DTestIntrospectionControl.cpp52 using WrapperType = nn::test_wrapper::Type; typedef
187 WrapperOperandType type0(WrapperType::TENSOR_FLOAT32, {2}); in createSimpleAddModel()
188 WrapperOperandType type1(WrapperType::INT32, {}); in createSimpleAddModel()
725 WrapperOperandType type0(WrapperType::TENSOR_FLOAT32, {2}); in createAddMulModel()
726 WrapperOperandType type1(WrapperType::INT32, {}); in createAddMulModel()
DTestPartitioningRandom.cpp109 using WrapperType = nn::test_wrapper::Type; typedef
414 const WrapperOperandType biasType(WrapperType::TENSOR_FLOAT32, { problemSize }); in makeRnnSpecialInput()
433 const WrapperOperandType permType(WrapperType::TENSOR_INT32, {2}); in makeTransposeSpecialInput()
699 const WrapperOperandType problemType(WrapperType::TENSOR_FLOAT32, { problemSize, problemSize }); in TEST_P()
700 const WrapperOperandType unknownDimensionsType(WrapperType::TENSOR_FLOAT32, { 0, 0 }); in TEST_P()
702 static const WrapperOperandType activationFunctionType(WrapperType::INT32, { }); in TEST_P()
DTestExecution.cpp49 using WrapperType = nn::test_wrapper::Type; typedef
434 static const WrapperOperandType tensorType(WrapperType::TENSOR_FLOAT32, { 1 }); in makeModel()
DTestPartitioning.cpp152 using WrapperType = ::android::nn::test_wrapper::Type; typedef
484 uint32_t addFloatOperand() { return addOperand(WrapperType::TENSOR_FLOAT32); } in addFloatOperand()
485 uint32_t addQuantOperand() { return addOperand(WrapperType::TENSOR_QUANT8_ASYMM); } in addQuantOperand()
489 uint32_t addOperand(WrapperType wrapperType) { in addOperand()
616 uint32_t operand = addOperand(WrapperType::INT32); in addIntOperand()
1689 uint32_t opndIn = model.addOperand(static_cast<WrapperType>(operandType)); in TEST_F()