Searched defs:GatherNDModel1 (Results 1 – 1 of 1) sorted by relevance
| /test/xts/acts/ai/nncore/opstest/src/ |
| D | gather_nd_test.cpp | 25 struct GatherNDModel1 { struct
26 const std::vector<int32_t> input_shape = {2, 3};
27 const std::vector<int32_t> indices_shape = {2, 2};
28 const std::vector<int32_t> output_shape = {2};
29 float indicesValue[2][2] = {{0, 0}, {1, 1}};
30 float inputValue[2][3] = {{-0.1, 0.3, 3.6}, {0.4, 0.5, -3.2}};
31 float outputValue[2] = {0};
33 OHNNOperandTest input = {OH_NN_FLOAT32, OH_NN_TENSOR, input_shape, inputValue, 6*sizeof(float)};
34 …NOperandTest indices = {OH_NN_FLOAT32, OH_NN_TENSOR, indices_shape, indicesValue, 4*sizeof(float)};
35 …OHNNOperandTest output = {OH_NN_FLOAT32, OH_NN_TENSOR, output_shape, outputValue, 2*sizeof(float)};
[all …]
|