Home
last modified time | relevance | path

Searched refs:dilationSize (Results 1 – 3 of 3) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Ddepthwise_conv2d_native_builder.cpp137 uint32_t dilationSize = tensor->GetElementCount(); in SetDilation() local
138 m_dilation.assign(pDilation, pDilation + dilationSize); in SetDilation()
H A Dconv2d_builder.cpp122 uint32_t dilationSize = tensor->GetElementCount(); in SetDilation() local
123 m_dilation.assign(pDilation, pDilation + dilationSize); in SetDilation()
H A Dconv2d_transpose_builder.cpp115 uint32_t dilationSize = tensor->GetElementCount(); in SetDilation() local
116 m_dilation.assign(pDilation, pDilation + dilationSize); in SetDilation()