Home
last modified time | relevance | path

Searched refs:weightDim (Results 1 – 6 of 6) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Ddepthwise_conv2d_native_pad_test.cpp78 std::vector<int32_t> weightDim = {2, 2, 2, 1}; in SetDepthwiseConv2dInput() local
84 inputsTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR); in SetDepthwiseConv2dInput()
428 std::vector<int32_t> weightDim = {1, 3, 3}; variable
435 inputsTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR);
464 std::vector<int32_t> weightDim = {2, 2, 2, 1}; variable
471 inTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR);
H A Dconv2d_padmode_test.cpp76 std::vector<int32_t> weightDim = {1, 2, 2, 1}; in SetConv2dInput() local
83 inputTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR); in SetConv2dInput()
460 std::vector<int32_t> weightDim = {1, 3, 1}; variable
466 inputTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR);
H A Dconv2d_tranpose_padmode_test.cpp90 std::vector<int32_t> weightDim = {1, 2, 2, 1}; in SetConv2dTransposeInput() local
95 inTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR); in SetConv2dTransposeInput()
453 std::vector<int32_t> weightDim = {1, 3, 1}; variable
459 inTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR);
H A Ddepthwise_conv2d_native_padmode_test.cpp75 std::vector<int32_t> weightDim = {2, 2, 2, 1}; in SetDepthwiseConv2dInput() local
81 inputTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR); in SetDepthwiseConv2dInput()
H A Dconv2d_transpose_pad_test.cpp94 std::vector<int32_t> weightDim = {1, 2, 2, 1}; in SetConv2dTransposeInput() local
100 tensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR); in SetConv2dTransposeInput()
H A Dconv2d_pad_test.cpp77 std::vector<int32_t> weightDim = {1, 2, 2, 1}; in SetConv2dInput() local
83 inputsTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR); in SetConv2dInput()