Searched refs:OH_NN_CONCAT_AXIS (Results 1 – 10 of 10) sorted by relevance
/ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
H A D | concat_two_inputs_test.cpp | 71 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 86 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 106 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 125 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 144 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 163 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 178 …hared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 200 …hared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 220 …hared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 237 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); [all …]
|
H A D | concat_three_inputs_test.cpp | 69 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 84 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 103 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 121 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 139 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 153 …hared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT32, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 175 …hared_ptr<NNTensor> tensor = TransToNNTensor(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 216 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS); 237 SetAxis(OH_NN_INT64, m_param_dim, nullptr, OH_NN_CONCAT_AXIS);
|
/ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
H A D | concat_builder.h | 48 {OH_NN_CONCAT_AXIS, &ConcatBuilder::SetAxis}
|
/ohos5.0/foundation/ai/neural_network_runtime/interfaces/kits/c/neural_network_runtime/ |
H A D | neural_network_runtime_type.h | 2836 OH_NN_CONCAT_AXIS = 10, enumerator
|
/ohos5.0/docs/zh-cn/application-dev/reference/apis-neural-network-runtime-kit/ |
H A D | neural__network__runtime__type_8h.md | 71 …BATCH_TO_SPACE_ND_BLOCKSIZE = 8, OH_NN_BATCH_TO_SPACE_ND_CROPS = 9, OH_NN_CONCAT_AXIS = 10, OH_NN_…
|
H A D | _neural_network_runtime.md | 72 …BATCH_TO_SPACE_ND_BLOCKSIZE = 8, OH_NN_BATCH_TO_SPACE_ND_CROPS = 9, OH_NN_CONCAT_AXIS = 10, OH_NN_… 784 | OH_NN_CONCAT_AXIS | 当张量作为Concat算子的axis参数时,使用本枚举值。 |
|
H A D | _neural_nework_runtime.md | 72 …BATCH_TO_SPACE_ND_BLOCKSIZE = 8, OH_NN_BATCH_TO_SPACE_ND_CROPS = 9, OH_NN_CONCAT_AXIS = 10, OH_NN_… 784 | OH_NN_CONCAT_AXIS | 当张量作为Concat算子的axis参数时,使用本枚举值。 |
|
/ohos5.0/docs/en/application-dev/reference/apis-neural-network-runtime-kit/ |
H A D | neural__network__runtime__type_8h.md | 71 …BATCH_TO_SPACE_ND_BLOCKSIZE = 8, OH_NN_BATCH_TO_SPACE_ND_CROPS = 9, OH_NN_CONCAT_AXIS = 10, OH_NN_…
|
H A D | _neural_network_runtime.md | 72 …BATCH_TO_SPACE_ND_BLOCKSIZE = 8, OH_NN_BATCH_TO_SPACE_ND_CROPS = 9, OH_NN_CONCAT_AXIS = 10, OH_NN_… 784 | OH_NN_CONCAT_AXIS | Used when the tensor is used as the **axis** parameter of the Concat operator…
|
/ohos5.0/foundation/ai/neural_network_runtime/example/deep_learning_framework/tflite/delegates/nnrt_delegate/ |
H A D | nnrt_op_builder.cpp | 177 mappingArgs.builder->AddScalarInt64Tensor(axis, OH_NN_CONCAT_AXIS); in AddConcatenationParams()
|