Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_op_builder.cpp35 NnrtOpBuilder::NnrtOpBuilder(const NnrtApi* nnrt, NnrtOpBuilderArgs& opBuilderArgs) in NnrtOpBuilder() argument
37 m_context(opBuilderArgs.context), in NnrtOpBuilder()
38 m_pTensorMapping(opBuilderArgs.pTensorMapping), in NnrtOpBuilder()
39 m_nnModel(opBuilderArgs.nnModel), in NnrtOpBuilder()
40 m_allowDynamicDimensions(opBuilderArgs.delegateOptions.allowDynamicDimensions) in NnrtOpBuilder()
46 for (int32_t i : TfLiteIntArrayView(opBuilderArgs.inputTensors)) { in NnrtOpBuilder()
48 …if (i != kTfLiteOptionalTensor && opBuilderArgs.context->tensors[i].allocation_type != kTfLiteMmap… in NnrtOpBuilder()
H A Dnnrt_delegate_kernel.cpp233 NnrtOpBuilderArgs opBuilderArgs = { in AddOpsAndTensors() local
240 NnrtOpBuilder builder(m_nnrt, opBuilderArgs); in AddOpsAndTensors()
H A Dnnrt_op_builder.h59 NnrtOpBuilder(const NnrtApi* nnrt, NnrtOpBuilderArgs& opBuilderArgs);