Searched defs:GatherNDModel2 (Results 1 – 1 of 1) sorted by relevance
| /test/xts/acts/ai/nncore/opstest/src/ |
| D | gather_nd_test.cpp | 44 struct GatherNDModel2 { struct
45 const std::vector<int32_t> input_shape = {2, 3};
46 const std::vector<int32_t> indices_shape = {2, 2};
47 const std::vector<int32_t> output_shape = {2};
48 float indicesValue[2][2] = {{1, 0}, {1, 0}};
49 float inputValue[2][3] = {{-0.1, 0.3, 3.6}, {0.4, 0.5, -3.2}};
50 float outputValue[2] = {0};
52 OHNNOperandTest input = {OH_NN_FLOAT32, OH_NN_TENSOR, input_shape, inputValue, 6*sizeof(float)};
53 …NOperandTest indices = {OH_NN_FLOAT32, OH_NN_TENSOR, indices_shape, indicesValue, 4*sizeof(float)};
54 …OHNNOperandTest output = {OH_NN_FLOAT32, OH_NN_TENSOR, output_shape, outputValue, 2*sizeof(float)};
[all …]
|