Searched defs:ReduceMaxModel5 (Results 1 – 1 of 1) sorted by relevance
| /test/xts/acts/ai/nncore/opstest/src/ |
| D | reducemax_test.cpp | 105 struct ReduceMaxModel5 { struct
106 const std::vector<int32_t> input_shape = {2, 2};
107 const std::vector<int32_t> output_shape = {1};
109 bool keepDimsValue[1] = {false};
110 float inputValue[2][2] = {{2, 3}, {6, 1}};
111 float axisValue[1] = {1};
112 float outputValue[1] = {0};
114 OHNNOperandTest input = {OH_NN_FLOAT32, OH_NN_TENSOR, input_shape, inputValue, 6*sizeof(float)};
115 OHNNOperandTest axis = {OH_NN_FLOAT32, OH_NN_TENSOR, {1}, axisValue, sizeof(float)};
116 …OHNNOperandTest output = {OH_NN_FLOAT32, OH_NN_TENSOR, output_shape, outputValue, 2*sizeof(float)};
[all …]
|