Searched defs:UnstackModel2 (Results 1 – 1 of 1) sorted by relevance
| /test/xts/acts/ai/nncore/opstest/src/ |
| D | unstack_test.cpp | 42 struct UnstackModel2 { struct
43 const std::vector<int32_t> input_shape = {3};
44 const std::vector<int32_t> output_shape = {1};
45 int64_t axisValue = 0;
46 float inputValue[3] = {1, 5, 6};
47 float outputValue[1] = {0};
49 OHNNOperandTest input = {OH_NN_FLOAT32, OH_NN_TENSOR, input_shape, inputValue, 3*sizeof(float)};
50 … OHNNOperandTest output = {OH_NN_FLOAT32, OH_NN_TENSOR, output_shape, outputValue, sizeof(float)};
51 OHNNOperandTest axis = {OH_NN_INT64, OH_NN_UNSTACK_AXIS, {1}, &axisValue, sizeof(int64_t)};
52 OHNNGraphArgs graphArgs = {.operationType = OH_NN_OPS_UNSTACK,
|