Searched defs:UnstackModel5 (Results 1 – 1 of 1) sorted by relevance
| /test/xts/acts/ai/nncore/opstest/src/ |
| D | unstack_test.cpp | 93 struct UnstackModel5 { struct
94 const std::vector<int32_t> tensor_shape = {};
95 int64_t axisValue = 0;
96 float* inputValue = {};
97 float* outputValue = {};
99 … OHNNOperandTest input = {OH_NN_FLOAT32, OH_NN_TENSOR, tensor_shape, inputValue, 0*sizeof(float)};
100 …OHNNOperandTest output = {OH_NN_FLOAT32, OH_NN_TENSOR, tensor_shape, outputValue, 0*sizeof(float)};
101 OHNNOperandTest axis = {OH_NN_INT64, OH_NN_UNSTACK_AXIS, {1}, &axisValue, sizeof(int64_t)};
102 OHNNGraphArgs graphArgs = {.operationType = OH_NN_OPS_UNSTACK,
|