| /ohos5.0/foundation/ai/neural_network_runtime/test/unittest/ops/ |
| H A D | unsqueeze_builder_test.cpp | 269 LiteGraphTensorPtr primitive = m_builder.GetPrimitive(); variable
289 LiteGraphTensorPtr primitive = m_builder.GetPrimitive(); variable
|
| H A D | avgpool_pad_test.cpp | 534 LiteGraphTensorPtr primitive = m_builder.GetPrimitive(); variable
570 LiteGraphTensorPtr primitive = m_builder.GetPrimitive(); variable
|
| H A D | conv2d_padmode_test.cpp | 550 LiteGraphTensorPtr primitive = m_builder.GetPrimitive(); variable
585 LiteGraphTensorPtr primitive = m_builder.GetPrimitive(); variable
|
| H A D | maxpool_pad_test.cpp | 535 LiteGraphTensorPtr primitive = m_builder.GetPrimitive(); variable
572 LiteGraphTensorPtr primitive = m_builder.GetPrimitive(); variable
|
| H A D | sub_builder_test.cpp | 294 LiteGraphTensorPtr primitive = m_builder.GetPrimitive(); variable
315 LiteGraphTensorPtr primitive = m_builder.GetPrimitive(); variable
|
| H A D | layernorm_builder_test.cpp | 440 LiteGraphPrimitvePtr primitive = m_layerNorm.GetPrimitive(); variable
459 LiteGraphPrimitvePtr primitive = m_layerNorm.GetPrimitive(); variable
|
| /ohos5.0/foundation/ai/neural_network_runtime/frameworks/native/neural_network_runtime/ops/ |
| H A D | leaky_relu_builder.cpp | 112 void* primitive = mindspore::lite::MindIR_Activation_CreatePrimitive(activationType, alpha, in GetPrimitive() local
|
| H A D | quant_dtype_cast_builder.cpp | 149 …void* primitive = mindspore::lite::MindIR_QuantDTypeCast_CreatePrimitive(*m_src_t, *m_dst_t, m_axi… in GetPrimitive() local
|
| H A D | scale_builder.cpp | 143 … void* primitive = mindspore::lite::MindIR_ScaleFusion_CreatePrimitive(*m_axis, m_activationType); in GetPrimitive() local
|
| H A D | constant_of_shape_builder.cpp | 131 void* primitive = mindspore::lite::MindIR_ConstantOfShape_CreatePrimitive(m_dataType, m_value); in GetPrimitive() local
|
| H A D | crop_builder.cpp | 134 void* primitive = mindspore::lite::MindIR_Crop_CreatePrimitive(m_axis, m_offset); in GetPrimitive() local
|
| H A D | exp_builder.cpp | 150 void* primitive = mindspore::lite::MindIR_ExpFusion_CreatePrimitive(m_base, m_scale, m_shift); in GetPrimitive() local
|
| H A D | unstack_builder.cpp | 127 void* primitive = mindspore::lite::MindIR_Unstack_CreatePrimitive(m_axis); in GetPrimitive() local
|
| H A D | depth_to_space_builder.cpp | 138 …void* primitive = mindspore::lite::MindIR_DepthToSpace_CreatePrimitive(m_blockSize, format, m_mode… in GetPrimitive() local
|
| H A D | pad_builder.cpp | 152 void* primitive = MindIR_PadFusion_CreatePrimitive(paddings, m_paddingMode, m_constantValue); in GetPrimitive() local
|
| H A D | range_builder.cpp | 151 … void* primitive = mindspore::lite::MindIR_Range_CreatePrimitive(dType, m_start, m_limit, m_delta); in GetPrimitive() local
|
| H A D | reduceL2_builder.cpp | 156 …void* primitive = mindspore::lite::MindIR_ReduceFusion_CreatePrimitive(m_keepDims, mode, m_reduceT… in GetPrimitive() local
|
| H A D | reduceall_builder.cpp | 156 …void* primitive = mindspore::lite::MindIR_ReduceFusion_CreatePrimitive(m_keepDims, mode, m_reduceT… in GetPrimitive() local
|
| H A D | reducemax_builder.cpp | 156 …void* primitive = mindspore::lite::MindIR_ReduceFusion_CreatePrimitive(m_keepDims, m_mode, m_reduc… in GetPrimitive() local
|
| H A D | reducemean_builder.cpp | 158 …void* primitive = mindspore::lite::MindIR_ReduceFusion_CreatePrimitive(m_keepDims, mode, m_reduceT… in GetPrimitive() local
|
| H A D | reducemin_builder.cpp | 156 …void* primitive = mindspore::lite::MindIR_ReduceFusion_CreatePrimitive(m_keepDims, m_mode, m_reduc… in GetPrimitive() local
|
| H A D | reduceprod_builder.cpp | 158 …void* primitive = mindspore::lite::MindIR_ReduceFusion_CreatePrimitive(m_keepDims, mode, m_reduceT… in GetPrimitive() local
|
| H A D | reducesum_builder.cpp | 156 …void* primitive = mindspore::lite::MindIR_ReduceFusion_CreatePrimitive(m_keepDims, m_mode, m_reduc… in GetPrimitive() local
|
| H A D | matmul_builder.cpp | 166 …auto primitive = mindspore::lite::MindIR_MatMulFusion_CreatePrimitive(m_transposeA, m_transposeB, … in GetPrimitive() local
|
| H A D | argmax_builder.cpp | 165 …void* primitive = mindspore::lite::MindIR_ArgMaxFusion_CreatePrimitive(m_axis, m_topK, m_keepDims,… in GetPrimitive() local
|