Home
last modified time | relevance | path

Searched defs:biasDim (Results 1 – 9 of 9) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/
H A Ddepthwise_conv2d_native_pad_test.cpp79 std::vector<int32_t> biasDim = {2}; in SetDepthwiseConv2dInput() local
429 std::vector<int32_t> biasDim = {2}; variable
465 std::vector<int32_t> biasDim = {2}; variable
H A Dbiasadd_test.cpp48 std::vector<int32_t> biasDim{3}; in SetBiasAddToallTensors() local
H A Dconv2d_padmode_test.cpp77 std::vector<int32_t> biasDim = {1}; in SetConv2dInput() local
461 std::vector<int32_t> biasDim = {1}; variable
H A Dconv2d_tranpose_padmode_test.cpp91 std::vector<int32_t> biasDim = {1}; in SetConv2dTransposeInput() local
454 std::vector<int32_t> biasDim = {1}; variable
H A Dfullconnection_test.cpp55 std::vector<int32_t> biasDim = {2}; in SetInputToAlltensor() local
H A Ddepthwise_conv2d_native_padmode_test.cpp76 std::vector<int32_t> biasDim = {2}; in SetDepthwiseConv2dInput() local
H A Dconv2d_pad_test.cpp78 std::vector<int32_t> biasDim = {1}; in SetConv2dInput() local
H A Dconv2d_transpose_pad_test.cpp95 std::vector<int32_t> biasDim = {1}; in SetConv2dTransposeInput() local
H A Dfullconnection_with_axis_test.cpp58 std::vector<int32_t> biasDim{2}; in SetInputToAlltensor() local