Home
last modified time | relevance | path

Searched refs:tensorDims (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.cpp401 const int32_t* tensorDims = nnTensor.dimensions; in TransposeDepthwiseTensor() local
405 …TF_LITE_ENSURE_STATUS(TransposeDims(m_context, tensorDims, tensorRank, DEPTHWISE_TRANSPOSE_AXISS, … in TransposeDepthwiseTensor()
411 … TransposeTensor(m_context, tensorIndex, tensorDims, reinterpret_cast<float*>(tensorData.data()))); in TransposeDepthwiseTensor()
414 …TransposeTensor(m_context, tensorIndex, tensorDims, reinterpret_cast<int32_t*>(tensorData.data()))… in TransposeDepthwiseTensor()
417 …TransposeTensor(m_context, tensorIndex, tensorDims, reinterpret_cast<int8_t*>(tensorData.data()))); in TransposeDepthwiseTensor()
420 …TransposeTensor(m_context, tensorIndex, tensorDims, reinterpret_cast<uint8_t*>(tensorData.data()))… in TransposeDepthwiseTensor()
440 int32_t* tensorDims = (m_allowDynamicDimensions && (tensor->allocation_type != kTfLiteMmapRo) && in ConstructNNTensor() local
448 tensorDims = const_cast<int32_t*>(&SCALAR_TENSOR_RANK); in ConstructNNTensor()
453 tensorDims = nullptr; in ConstructNNTensor()
458 nnTensor.dimensions = tensorDims; in ConstructNNTensor()
H A Dnnrt_delegate_kernel.cpp300 int32_t* tensorDims = reinterpret_cast<int32_t*>(tensor->dims->data); in ConvertTensorTypeToNn() local
301 if (tensorDims == nullptr) { in ConvertTensorTypeToNn()
310 *tensorDims = SCALAR_RANK; in ConvertTensorTypeToNn()
315 nnTensor.dimensions = tensorDims; in ConvertTensorTypeToNn()
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dneural_network_runtime.cpp340 auto tensorDims = mindspore::lite::MindIR_Tensor_GetDims(tensor); in ParseInputDimsFromExtensions() local
341 size_t inputDimSize = tensorDims.size(); in ParseInputDimsFromExtensions()