Home
last modified time | relevance | path

Searched refs:tensorData (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.cpp357 void* tensorData = tensor->data.data; in AddTensor() local
375 tensorData = static_cast<void*>(depthwiseTensorData.data()); in AddTensor()
388 tensorData, tensor->bytes); in AddTensor()
399 std::vector<int32_t>& weightDims, std::vector<int8_t>& tensorData) in TransposeDepthwiseTensor() argument
411 … TransposeTensor(m_context, tensorIndex, tensorDims, reinterpret_cast<float*>(tensorData.data()))); in TransposeDepthwiseTensor()
414 …ransposeTensor(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 …ransposeTensor(m_context, tensorIndex, tensorDims, reinterpret_cast<uint8_t*>(tensorData.data()))); in TransposeDepthwiseTensor()
H A Dnnrt_utils.h117 const T* tensorData = reinterpret_cast<T*>(tensor->data.data); in TransposeTensor() local
128 *(transposeTensor + newPos) = *(tensorData + orgPos); in TransposeTensor()
H A Dnnrt_op_builder.h282 std::vector<int8_t>& tensorData);