Home
last modified time | relevance | path

Searched refs:mappingArgs (Results 1 – 4 of 4) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_op_builder.cpp58 mappingArgs.context->AddTensors(mappingArgs.context, 1, &biasIndex); in AddZerosBias()
59 TfLiteTensor* biasTensor = &mappingArgs.context->tensors[biasIndex]; in AddZerosBias()
60 const auto inputType = mappingArgs.context->tensors[inputId].type; in AddZerosBias()
72 mappingArgs.context->ResizeTensor(mappingArgs.context, biasTensor, biasShape); in AddZerosBias()
157 …(mappingArgs.node->inputs->size == 3) && (mappingArgs.node->inputs->data[2] != kTfLiteOptionalTens… in AddFullConnectedParams()
177 mappingArgs.builder->AddScalarInt64Tensor(axis, OH_NN_CONCAT_AXIS); in AddConcatenationParams()
195 int32_t inputIndex = mappingArgs.node->inputs->data[0]; in AddQuantizeParams()
199 int32_t outputIndex = mappingArgs.node->outputs->data[0]; in AddQuantizeParams()
210 mappingArgs.builder->AddScalarInt64Tensor(axis, OH_NN_STACK_AXIS); in AddPackParams()
250 if (mappingArgs.node->inputs->size == 1) { in AddReshapeParams()
[all …]
H A Dnnrt_op_builder.h119 TfLiteStatus AddOpFuncParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
226 mappingArgs.builder->AddScalarInt8Tensor(padding, nnTensorType); in AddPadParamsInOperator()
238 TfLiteStatus AddAvgPoolingParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
240 TfLiteStatus AddMaxPoolingParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
244 TfLiteStatus AddConv2DParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
248 TfLiteStatus AddSoftmaxParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
250 TfLiteStatus AddQuantizeParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
252 TfLiteStatus AddPackParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
254 TfLiteStatus AddPadParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
258 TfLiteStatus AddReshapeParams(const NnrtOpMappingArgs& mappingArgs, int32_t builtinCode);
[all …]
H A Dnnrt_delegate_kernel.cpp159 …LiteStatus NnrtDelegateKernel::Map(const int32_t builtinCode, const NnrtOpMappingArgs& mappingArgs, in Map() argument
168 TfLiteStatus retValue = mappingArgs.builder->AddOpFuncParams(mappingArgs, builtinCode); in Map()
H A Dnnrt_delegate_kernel.h65 …TfLiteStatus Map(int32_t builtinCode, const NnrtOpMappingArgs& mappingArgs, int32_t& nnOpType) con…