| /ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/ |
| H A D | nnrt_utils.h | 98 extern TfLiteStatus TransposeDims(TfLiteContext* context, const int32_t* dims, uint32_t dimCount,
104 extern TfLiteStatus GetTensorSize(TfLiteContext* context, const int32_t* dims, int32_t dimCount, in…
110 TfLiteStatus TransposeTensor(TfLiteContext* context, int32_t tensorIndex, const int32_t* dims, in TransposeTensor() argument
113 TF_LITE_ENSURE_EQ(context, dims != nullptr, true); in TransposeTensor()
118 const int32_t batch = dims[DEPTHWISE_WEIGHT_BATCH_DIMENSION]; in TransposeTensor()
119 const int32_t height = dims[DEPTHWISE_WEIGHT_HEIGHT_DIMENSION]; in TransposeTensor()
120 const int32_t width = dims[DEPTHWISE_WEIGHT_WIDTH_DIMENSION]; in TransposeTensor()
121 const int32_t channel = dims[DEPTHWISE_WEIGHT_CHANNEL_DIMENSION]; in TransposeTensor()
|
| H A D | nnrt_utils.cpp | 142 TfLiteStatus TransposeDims(TfLiteContext* context, const int32_t* dims, uint32_t dimCount, in TransposeDims() argument
145 TF_LITE_ENSURE_EQ(context, dims != nullptr, true); in TransposeDims()
157 weightDims.emplace_back(*(dims + axis)); in TransposeDims()
163 TfLiteStatus GetTensorSize(TfLiteContext* context, const int32_t* dims, int32_t dimCount, int64_t& … in GetTensorSize() argument
165 TF_LITE_ENSURE_EQ(context, dims != nullptr, true); in GetTensorSize()
175 if (*(dims + i) <= 0) { in GetTensorSize()
179 tensorSize *= *(dims + i); in GetTensorSize()
|
| H A D | nnrt_op_builder.cpp | 162 …const int32_t numUnits = mappingArgs.context->tensors[filterTensorId].dims->data[0]; // bias chan… in AddFullConnectedParams()
437 uint32_t tensorRank = static_cast<uint32_t>(tensor->dims->size); in ConstructNNTensor()
443 tensor->dims->data; in ConstructNNTensor()
|
| H A D | nnrt_delegate_kernel.cpp | 299 uint32_t tensorRank = static_cast<uint32_t>(tensor->dims->size); in ConvertTensorTypeToNn()
300 int32_t* tensorDims = reinterpret_cast<int32_t*>(tensor->dims->data); in ConvertTensorTypeToNn()
|
| /ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/tools/ |
| H A D | utils.cpp | 135 TfLiteIntArray* outputsDims = interpreter->tensor(output_index)->dims; in PrintResult()
174 TfLiteIntArray* outputDims = interpreter->tensor(output)->dims; in AnalysisResults()
226 TfLiteIntArray* dims = interpreter->tensor(input)->dims; in ImportData() local
228 wantedimageInfo.height = dims->data[1]; in ImportData()
229 wantedimageInfo.width = dims->data[WEIGHT_DIMENSION]; in ImportData()
230 wantedimageInfo.channels = (dims->size > CHANNEL_DIMENSION) ? dims->data[CHANNEL_DIMENSION] : 1; in ImportData()
256 bool IsEqualShape(int tensorIndex, const std::vector<int>& dims, std::unique_ptr<tflite::Interprete… in IsEqualShape() argument
259 for (int i = 0; i < tensor->dims->size; ++i) { in IsEqualShape()
260 if (tensor->dims->data[i] != dims[i]) { in IsEqualShape()
|
| /ohos5.0/docs/zh-cn/device-dev/reference/hdi-apis/nnrt/ |
| H A D | _tile_fusion_v10.md | 32 | long[] [dims](#dims) | 它和multiples功能相同，1维tensor，指定各个维度拷贝的次数。其长度m不小于x的维数n。 |
38 ### dims subsection
41 long [] TileFusion::dims
|
| H A D | _tile_fusion_v20.md | 32 | long[] [dims](#dims) | 它和multiples功能相同，1维tensor，指定各个维度拷贝的次数。其长度m不小于x的维数n。 |
38 ### dims subsection
41 long [] TileFusion::dims
|
| H A D | _tensor_v10.md | 22 | int[] [dims](#dims) | 张量维度数组。 |
51 ### dims subsection
54 int [] Tensor::dims
|
| H A D | _tensor_v20.md | 22 | int[] [dims](#dims) | 张量维度数组。 |
51 ### dims subsection
54 int [] Tensor::dims
|
| /ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v1_0/cpp_target/nnrt/v1_0/ |
| H A D | model_types.cpp.txt | 83 if (!WritePodArray(data, dataBlock.dims)) {
84 HDF_LOGE("%{public}s: failed to write dataBlock.dims", __func__);
123 if (!ReadPodArray(data, dataBlock.dims)) {
124 HDF_LOGE("%{public}s: failed to read dataBlock.dims", __func__);
|
| H A D | model_types.h.txt | 60 std::vector<int32_t> dims;
|
| /ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v2_0/cpp_target/nnrt/v2_0/ |
| H A D | model_types.cpp.txt | 83 if (!WritePodArray(data, dataBlock.dims)) {
84 HDF_LOGE("%{public}s: failed to write dataBlock.dims", __func__);
123 if (!ReadPodArray(data, dataBlock.dims)) {
124 HDF_LOGE("%{public}s: failed to read dataBlock.dims", __func__);
|
| H A D | model_types.h.txt | 60 std::vector<int32_t> dims;
|
| /ohos5.0/drivers/interface/nnrt/v1_0/ |
| H A D | ModelTypes.idl | 23 int[] dims;
|
| H A D | NodeAttrTypes.idl | 299 long[] dims;
|
| /ohos5.0/drivers/interface/nnrt/v2_0/ |
| H A D | ModelTypes.idl | 23 int[] dims;
|
| H A D | NodeAttrTypes.idl | 300 long[] dims;
|
| /ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v1_0/foo/nnrt/v1_0/ |
| H A D | ModelTypes.idl | 23 int[] dims;
|
| H A D | NodeAttrTypes.idl | 299 long[] dims;
|
| /ohos5.0/foundation/ability/idl_tool/test/hdi_unittest/nnrt_v2_0/foo/nnrt/v2_0/ |
| H A D | ModelTypes.idl | 23 int[] dims;
|
| H A D | NodeAttrTypes.idl | 300 long[] dims;
|
| /ohos5.0/foundation/ai/neural_network_runtime/example/drivers/nnrt/v1_0/hdi_cpu_service/src/ |
| H A D | nnrt_device_service.cpp | 340 schemaTensor->dims = tensor.dims; in TransTensor()
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/inner_kits/ |
| H A D | neural_network_runtime_inner_test.cpp | 163 int32_t *dims = &dim; variable
171 …mindspore::lite::TensorPtr ptr2 = mindspore::lite::MindIR_Tensor_Create(&a, data_type, dims, dims_…
|
| /ohos5.0/foundation/ai/neural_network_runtime/example/drivers/nnrt/v2_0/hdi_cpu_service/src/ |
| H A D | nnrt_device_service.cpp | 381 schemaTensor->dims = tensor.dims; in TransTensor()
|
| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_runtime_test/ |
| H A D | neural_network_runtime_test.cpp | 1270 int32_t dims[2] = {3, 4}; variable
1271 m_tensor = {OH_NN_FLOAT32, 2, dims, nullptr, OH_NN_TENSOR};
1300 int32_t dims[2] = {3, 4}; variable
1301 m_tensor = {OH_NN_FLOAT32, 2, dims, nullptr, OH_NN_TENSOR};
1362 int32_t dims[2] = {3, 4}; variable
1363 m_tensor = {OH_NN_FLOAT32, 2, dims, nullptr, OH_NN_TENSOR};
1392 int32_t dims[2] = {3, 4}; variable
1393 m_tensor = {OH_NN_FLOAT32, 2, dims, nullptr, OH_NN_TENSOR};
2419 int32_t dims[2] = {3, 4}; variable
2420 m_tensor = {OH_NN_FLOAT32, 2, dims, nullptr, OH_NN_TENSOR};
|