Searched defs:DepthToSpaceModel1 (Results 1 – 1 of 1) sorted by relevance
| /test/xts/acts/ai/nncore/opstest/src/ |
| D | depth_to_space_test.cpp | 25 struct DepthToSpaceModel1 { struct
26 const std::vector<int32_t> input_shape = {1, 4, 4, 4};
27 const std::vector<int32_t> output_shape = {1, 1, 8, 8};
28 std::vector<int64_t> blockSizeValue = {2};
29 … float inputValue[1][4][4][4] = {{{{1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15, 16}},
33 float outputValue[1][1][8][8] = {0};
35 … OHNNOperandTest input = {OH_NN_FLOAT32, OH_NN_TENSOR, input_shape, inputValue, 64*sizeof(float)};
36 …HNNOperandTest output = {OH_NN_FLOAT32, OH_NN_TENSOR, output_shape, outputValue, 64*sizeof(float)};
37 … blockSize = {OH_NN_INT64, OH_NN_DEPTH_TO_SPACE_BLOCK_SIZE, {1}, &blockSizeValue, sizeof(int64_t)};
38 OHNNGraphArgs graphArgs = {.operationType = OH_NN_OPS_DEPTH_TO_SPACE,
|