Searched defs:ReduceMaxModel4 (Results 1 – 1 of 1) sorted by relevance
| /test/xts/acts/ai/nncore/opstest/src/ |
| D | reducemax_test.cpp | 85 struct ReduceMaxModel4 { struct
86 const std::vector<int32_t> input_shape = {2, 3};
87 const std::vector<int32_t> output_shape = {1, 1};
89 bool keepDimsValue[1] = {true};
90 float inputValue[2][3] = {{1, 2, 3}, {4, 0, 6}};
91 float axisValue[2] = {0, 1};
92 float outputValue[1][1] = {0};
94 OHNNOperandTest input = {OH_NN_FLOAT32, OH_NN_TENSOR, input_shape, inputValue, 6*sizeof(float)};
95 OHNNOperandTest axis = {OH_NN_FLOAT32, OH_NN_TENSOR, {2}, axisValue, 2*sizeof(float)};
96 … OHNNOperandTest output = {OH_NN_FLOAT32, OH_NN_TENSOR, output_shape, outputValue, sizeof(float)};
[all …]
|