Searched defs:ReduceSumModel3 (Results 1 – 1 of 1) sorted by relevance
| /test/xts/acts/ai/nncore/opstest/src/ |
| D | reducesum_test.cpp | 69 struct ReduceSumModel3 { struct
70 const std::vector<int32_t> input_shape = {2, 3};
71 const std::vector<int32_t> output_shape = {2, 3};
73 bool keepDimsValue[1] = {false};
74 bool reduceToEndValue[1] = {true};
75 float inputValue[2][3] = {{1, 2, 3}, {4, 5, 6}};
76 float axisValue = {};
77 float outputValue[2][3] = {0};
79 OHNNOperandTest input = {OH_NN_FLOAT32, OH_NN_TENSOR, input_shape, inputValue, 6*sizeof(float)};
80 OHNNOperandTest axis = {OH_NN_FLOAT32, OH_NN_TENSOR, {}, &axisValue, 0*sizeof(float)};
[all …]
|