Searched refs:weightDim (Results 1 - 6 of 6) sorted by relevance
/foundation/ai/neural_network_runtime/test/unittest/ops/ |
H A D | depthwise_conv2d_native_pad_test.cpp | 78 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}; in HWTEST_F() local 435 inputsTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR); in HWTEST_F() 464 std::vector<int32_t> weightDim = {2, 2, 2, 1}; in HWTEST_F() local 471 inTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
|
H A D | conv2d_padmode_test.cpp | 76 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}; in HWTEST_F() local 466 inputTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
|
H A D | depthwise_conv2d_native_padmode_test.cpp | 75 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 D | conv2d_tranpose_padmode_test.cpp | 90 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}; in HWTEST_F() local 459 inTensor = TransToNNTensor(OH_NN_FLOAT32, weightDim, nullptr, OH_NN_TENSOR); in HWTEST_F()
|
H A D | conv2d_transpose_pad_test.cpp | 94 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 D | conv2d_pad_test.cpp | 77 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()
|
Completed in 6 milliseconds