Searched refs:weightDims (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 D | nnrt_op_builder.cpp | 356 std::vector<int32_t> weightDims; in AddTensor() local
370 …TfLiteStatus retCode = TransposeDepthwiseTensor(tensorIndex, nnTensor, weightDims, depthwiseTensor… in AddTensor()
399 std::vector<int32_t>& weightDims, std::vector<int8_t>& tensorData) in TransposeDepthwiseTensor() argument
405 …RE_STATUS(TransposeDims(m_context, tensorDims, tensorRank, DEPTHWISE_TRANSPOSE_AXISS, weightDims)); in TransposeDepthwiseTensor()
406 nnTensor.dimensions = weightDims.data(); in TransposeDepthwiseTensor()
|
| H A D | nnrt_utils.cpp | 143 std::vector<int32_t> destAxis, std::vector<int32_t>& weightDims) in TransposeDims() argument
157 weightDims.emplace_back(*(dims + axis)); in TransposeDims()
|
| H A D | nnrt_utils.h | 99 std::vector<int32_t> destAxis, std::vector<int32_t>& weightDims);
|