Home
last modified time | relevance | path

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

/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/
H A Dnnrt_op_builder.cpp125 …mappingArgs.builder->AddVectorInt64Tensor(kernel.data(), kernel.size(), OH_NN_AVG_POOL_KERNEL_SIZE… in AddAvgPoolingParams()
126 mappingArgs.builder->AddVectorInt64Tensor(stride.data(), stride.size(), OH_NN_AVG_POOL_STRIDE); in AddAvgPoolingParams()
143 …mappingArgs.builder->AddVectorInt64Tensor(kernel.data(), kernel.size(), OH_NN_MAX_POOL_KERNEL_SIZE… in AddMaxPoolingParams()
144 mappingArgs.builder->AddVectorInt64Tensor(stride.data(), stride.size(), OH_NN_MAX_POOL_STRIDE); in AddMaxPoolingParams()
272 mappingArgs.builder->AddVectorInt64Tensor(stride.data(), stride.size(), OH_NN_CONV2D_STRIDES); in AddConv2DParams()
273 …mappingArgs.builder->AddVectorInt64Tensor(dilation.data(), dilation.size(), OH_NN_CONV2D_DILATION); in AddConv2DParams()
291 TF_LITE_ENSURE_STATUS(mappingArgs.builder->AddVectorInt64Tensor(stride.data(), stride.size(), in AddDepthwiseConv2DParams()
293 … TF_LITE_ENSURE_STATUS(mappingArgs.builder->AddVectorInt64Tensor(dilation.data(), dilation.size(), in AddDepthwiseConv2DParams()
H A Dnnrt_op_builder.h89 …TfLiteStatus AddVectorInt64Tensor(const int64_t* values, uint32_t numValues, OH_NN_TensorType nnTe… in AddVectorInt64Tensor() function