Searched defs:LogSoftmaxModel3 (Results 1 – 1 of 1) sorted by relevance
| /test/xts/acts/ai/nncore/opstest/src/ |
| D | log_softmax_test.cpp | 57 struct LogSoftmaxModel3 { struct
58 const std::vector<int32_t> tensor_shape = {4, 2};
59 int64_t axisValue[1] = {2};
60 float inputValue[4][2] = {{1, 2}, {3, 4}, {5, 6}, {7, 8}};
61 float outputValue[4][2] = {0};
63 … OHNNOperandTest input = {OH_NN_FLOAT32, OH_NN_TENSOR, tensor_shape, inputValue, 8*sizeof(float)};
64 …OHNNOperandTest output = {OH_NN_FLOAT32, OH_NN_TENSOR, tensor_shape, outputValue, 8*sizeof(float)};
65 OHNNOperandTest axis = {OH_NN_INT64, OH_NN_LOG_SOFTMAX_AXIS, {1}, axisValue, sizeof(int64_t)};
66 OHNNGraphArgs graphArgs = {.operationType = OH_NN_OPS_LOG_SOFTMAX,
|