Home
last modified time | relevance | path

Searched refs:IN_CHANNEL_INDEX (Results 1 – 2 of 2) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dconv2d_builder.cpp31 static constexpr int IN_CHANNEL_INDEX = 3; variable
69 m_inChannel = weightShape[IN_CHANNEL_INDEX]; in SetChannel()
H A Dconv2d_transpose_builder.cpp31 static constexpr int IN_CHANNEL_INDEX = 3; variable
63 m_inChannel = weightShape[IN_CHANNEL_INDEX]; in SetInput()