Searched defs:AllModel2 (Results 1 – 1 of 1) sorted by relevance
| /test/xts/acts/ai/nncore/opstest/src/ |
| D | all_test.cpp | 44 struct AllModel2 { struct
45 const std::vector<int32_t> input_shape = {2, 3};
46 const std::vector<int32_t> output_shape = {2, 1};
47 std::vector<int64_t> keepDimsValue = {1};
48 bool inputValue[2][3] = {{true, false, true}, {true, true, true}};
49 int64_t axisValue[1] = {1};
50 bool outputValue[2][1] = {false};
52 OHNNOperandTest input = {OH_NN_BOOL, OH_NN_TENSOR, input_shape, inputValue, 6*sizeof(float)};
53 OHNNOperandTest axis = {OH_NN_INT64, OH_NN_TENSOR, {1}, axisValue, sizeof(int64_t)};
54 OHNNOperandTest output = {OH_NN_BOOL, OH_NN_TENSOR, output_shape, outputValue, 2*sizeof(float)};
[all …]
|