Home
last modified time | relevance | path

Searched defs:inputShape (Results 1 – 7 of 7) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/
H A Dlayernorm_builder.cpp145 auto inputShape = allTensors[inputsIndex[INPUT_X]]->GetDimensions(); in Build() local
182 auto inputShape = allTensors[inputsIndex[INPUT_X]]->GetDimensions(); in ValidateGammaAndBetaShape() local
H A Ddepthwise_conv2d_native_builder.cpp220 auto inputShape = allTensors[inputsIndex[INPUT_X]]->GetDimensions(); in Build() local
/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/label_classify/
H A Dlabel_classify.h39 string inputShape = ""; member
H A Dlabel_classify.cpp147 for (const auto& inputShape : neededInputShapes) { in PrepareModel() local
/ohos5.0/foundation/ai/neural_network_runtime/example/drivers/nnrt/v2_0/hdi_cpu_service/src/
H A Dprepared_model_service.cpp149 for (auto inputShape : m_inputDims) { in GetInputDimRanges() local
232 auto inputShape = graph->allTensors[i]->dims; in Compile() local
/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/tools/
H A Dutils.cpp106 for (const auto& inputShape : userInputShapes) { in FilterDynamicInputs() local
/ohos5.0/foundation/ai/neural_network_runtime/example/drivers/nnrt/v1_0/hdi_cpu_service/src/
H A Dprepared_model_service.cpp192 auto inputShape = graph->allTensors[i]->dims; in Compile() local