Searched defs:UnstackModel3 (Results 1 – 1 of 1) sorted by relevance
| /test/xts/acts/ai/nncore/opstest/src/ |
| D | unstack_test.cpp | 59 struct UnstackModel3 { struct
60 const std::vector<int32_t> input_shape = {3, 2};
61 const std::vector<int32_t> output_shape = {3, 1};
62 int64_t axisValue = 1;
63 float inputValue[3][2] = {{1, 2}, {3, 4}, {5, 6}};
64 float outputValue[3][1] = {0};
66 OHNNOperandTest input = {OH_NN_FLOAT32, OH_NN_TENSOR, input_shape, inputValue, 6*sizeof(float)};
67 …OHNNOperandTest output = {OH_NN_FLOAT32, OH_NN_TENSOR, output_shape, outputValue, 3*sizeof(float)};
68 OHNNOperandTest axis = {OH_NN_INT64, OH_NN_UNSTACK_AXIS, {1}, &axisValue, sizeof(int64_t)};
69 OHNNGraphArgs graphArgs = {.operationType = OH_NN_OPS_UNSTACK,
|