Home
last modified time | relevance | path

Searched refs:tensorType (Results 1 – 12 of 12) sorted by relevance

/ohos5.0/foundation/ai/neural_network_runtime/interfaces/kits/c/neural_network_runtime/
H A Dneural_network_runtime.h229 …eturnCode OH_NNModel_SetTensorType(OH_NNModel *model, uint32_t index, OH_NN_TensorType tensorType);
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/
H A Dnn_tensor.h58 OH_NN_ReturnCode SetTensorType(OH_NN_TensorType tensorType);
H A Dinner_model.h41 OH_NN_ReturnCode SetTensorType(uint32_t index, OH_NN_TensorType tensorType);
H A Dneural_network_runtime.cpp198 …ReturnCode OH_NNModel_SetTensorType(OH_NNModel* model, uint32_t index, OH_NN_TensorType tensorType) in OH_NNModel_SetTensorType() argument
205 if (!Validation::ValidateTensorType(tensorType)) { in OH_NNModel_SetTensorType()
211 OH_NN_ReturnCode returnCode = innerModel->SetTensorType((uint32_t)(index), tensorType); in OH_NNModel_SetTensorType()
H A Dnn_tensor.cpp224 OH_NN_ReturnCode NNTensor::SetTensorType(OH_NN_TensorType tensorType) in SetTensorType() argument
226 m_type = tensorType; in SetTensorType()
H A Dinner_model.cpp228 OH_NN_ReturnCode InnerModel::SetTensorType(uint32_t index, OH_NN_TensorType tensorType) in SetTensorType() argument
241 OH_NN_ReturnCode returnCode = tensor->SetTensorType(tensorType); in SetTensorType()
/ohos5.0/docs/zh-cn/application-dev/reference/apis-neural-network-runtime-kit/
H A Dneural__network__runtime_8h.md37 …index, [OH_NN_TensorType](_neural_network_runtime.md#oh_nn_tensortype) tensorType) | 设置张量的类型，参考[OH…
H A D_neural_network_runtime.md138 …nmodel) \*model, uint32_t index, [OH_NN_TensorType](#oh_nn_tensortype) tensorType) | 设置张量的类型，参考[OH…
2287 …turnCode OH_NNModel_SetTensorType (OH_NNModel *model, uint32_t index, OH_NN_TensorType tensorType )
2302 | tensorType | 张量类型。 |
H A D_neural_nework_runtime.md138 …nmodel) \*model, uint32_t index, [OH_NN_TensorType](#oh_nn_tensortype) tensorType) | 设置张量的类型，参考[OH…
2287 …turnCode OH_NNModel_SetTensorType (OH_NNModel *model, uint32_t index, OH_NN_TensorType tensorType )
2302 | tensorType | 张量类型。 |
/ohos5.0/docs/en/application-dev/reference/apis-neural-network-runtime-kit/
H A Dneural__network__runtime_8h.md37 …index, [OH_NN_TensorType](_neural_network_runtime.md#oh_nn_tensortype) tensorType) | Sets the tens…
H A D_neural_network_runtime.md138 …nmodel) \*model, uint32_t index, [OH_NN_TensorType](#oh_nn_tensortype) tensorType) | Sets the tens…
2287 …turnCode OH_NNModel_SetTensorType (OH_NNModel *model, uint32_t index, OH_NN_TensorType tensorType )
2302 | tensorType | Tensor type.|
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/components/v1_0/neural_network_runtime_test/
H A Dneural_network_runtime_test.cpp3032 OH_NN_TensorType tensorType = OH_NN_REDUCE_MIN_KEEP_DIMS; variable
3034 OH_NN_ReturnCode ret = OH_NNModel_SetTensorType(nullptr, index, tensorType);
3047 OH_NN_TensorType tensorType = OH_NN_REDUCE_MIN_COEFF; variable
3049 OH_NN_ReturnCode ret = OH_NNModel_SetTensorType(model, index, tensorType);