Home
last modified time | relevance | path

Searched defs:weightShape (Results 1 – 3 of 3) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dconv2d_builder.cpp63 auto weightShape = allTensors[inputsIndex[CONV2D_INPUT_WEIGHT]]->GetDimensions(); in SetChannel() local
79 auto weightShape = allTensors[inputsIndex[CONV2D_INPUT_WEIGHT]]->GetDimensions(); in SetKernelSize() local
H A Dconv2d_transpose_builder.cpp57 auto weightShape = allTensors[inputsIndex[INPUT_WEIGHT]]->GetDimensions(); in SetInput() local
72 auto weightShape = allTensors[inputsIndex[INPUT_WEIGHT]]->GetDimensions(); in SetKernelSize() local
H A Ddepthwise_conv2d_native_builder.cpp91 auto weightShape = allTensors[inputsIndex[INPUT_WEIGHT]]->GetDimensions(); in SetKernelSize() local