forked from mindspore-Ecosystem/mindspore
!32462 【MSLITE】white list ops
Merge pull request !32462 from fangzhou0329/fz_dev
This commit is contained in:
Gitee
commit
6d0faa556c
|
|
@ -31,6 +31,217 @@ set(BENCHMARK_TRAIN_NAME benchmark_train)
|
|||
set(BENCHMARK_TRAIN_ROOT_DIR ${RUNTIME_PKG_NAME}/tools/benchmark_train)
|
||||
file(GLOB JPEGTURBO_LIB_LIST ${jpeg_turbo_LIBPATH}/*.so)
|
||||
|
||||
function(__install_white_list_ops)
|
||||
install(FILES
|
||||
${TOP_DIR}/mindspore/core/ops/abs.h
|
||||
${TOP_DIR}/mindspore/core/ops/adam.h
|
||||
${TOP_DIR}/mindspore/core/ops/addn.h
|
||||
${TOP_DIR}/mindspore/core/ops/all.h
|
||||
${TOP_DIR}/mindspore/core/ops/apply_momentum.h
|
||||
${TOP_DIR}/mindspore/core/ops/assert.h
|
||||
${TOP_DIR}/mindspore/core/ops/assign.h
|
||||
${TOP_DIR}/mindspore/core/ops/assign_add.h
|
||||
${TOP_DIR}/mindspore/core/ops/audio_spectrogram.h
|
||||
${TOP_DIR}/mindspore/core/ops/batch_norm.h
|
||||
${TOP_DIR}/mindspore/core/ops/batch_to_space.h
|
||||
${TOP_DIR}/mindspore/core/ops/batch_to_space_nd.h
|
||||
${TOP_DIR}/mindspore/core/ops/bias_add.h
|
||||
${TOP_DIR}/mindspore/core/ops/binary_cross_entropy.h
|
||||
${TOP_DIR}/mindspore/core/ops/broadcast.h
|
||||
${TOP_DIR}/mindspore/core/ops/cast.h
|
||||
${TOP_DIR}/mindspore/core/ops/ceil.h
|
||||
${TOP_DIR}/mindspore/core/ops/clip.h
|
||||
${TOP_DIR}/mindspore/core/ops/concat.h
|
||||
${TOP_DIR}/mindspore/core/ops/attention.h
|
||||
${TOP_DIR}/mindspore/core/ops/cos.h
|
||||
${TOP_DIR}/mindspore/core/ops/constant_of_shape.h
|
||||
${TOP_DIR}/mindspore/core/ops/crop.h
|
||||
${TOP_DIR}/mindspore/core/ops/custom_extract_features.h
|
||||
${TOP_DIR}/mindspore/core/ops/custom_normalize.h
|
||||
${TOP_DIR}/mindspore/core/ops/custom_predict.h
|
||||
${TOP_DIR}/mindspore/core/ops/depend.h
|
||||
${TOP_DIR}/mindspore/core/ops/depth_to_space.h
|
||||
${TOP_DIR}/mindspore/core/ops/detection_post_process.h
|
||||
${TOP_DIR}/mindspore/core/ops/dropout.h
|
||||
${TOP_DIR}/mindspore/core/ops/elu.h
|
||||
${TOP_DIR}/mindspore/core/ops/eltwise.h
|
||||
${TOP_DIR}/mindspore/core/ops/equal.h
|
||||
${TOP_DIR}/mindspore/core/ops/expand_dims.h
|
||||
${TOP_DIR}/mindspore/core/ops/fake_quant_with_min_max_vars.h
|
||||
${TOP_DIR}/mindspore/core/ops/fake_quant_with_min_max_vars_per_channel.h
|
||||
${TOP_DIR}/mindspore/core/ops/fake_quant_with_min_max_vars.h
|
||||
${TOP_DIR}/mindspore/core/ops/fft_real.h
|
||||
${TOP_DIR}/mindspore/core/ops/fft_imag.h
|
||||
${TOP_DIR}/mindspore/core/ops/flatten.h
|
||||
${TOP_DIR}/mindspore/core/ops/floor.h
|
||||
${TOP_DIR}/mindspore/core/ops/floor_div.h
|
||||
${TOP_DIR}/mindspore/core/ops/floor_mod.h
|
||||
${TOP_DIR}/mindspore/core/ops/fill.h
|
||||
${TOP_DIR}/mindspore/core/ops/fused_batch_norm.h
|
||||
${TOP_DIR}/mindspore/core/ops/gather.h
|
||||
${TOP_DIR}/mindspore/core/ops/gather_nd.h
|
||||
${TOP_DIR}/mindspore/core/ops/greater.h
|
||||
${TOP_DIR}/mindspore/core/ops/greater_equal.h
|
||||
${TOP_DIR}/mindspore/core/ops/hashtable_lookup.h
|
||||
${TOP_DIR}/mindspore/core/ops/instance_norm.h
|
||||
${TOP_DIR}/mindspore/core/ops/leaky_relu.h
|
||||
${TOP_DIR}/mindspore/core/ops/less.h
|
||||
${TOP_DIR}/mindspore/core/ops/less_equal.h
|
||||
${TOP_DIR}/mindspore/core/ops/log.h
|
||||
${TOP_DIR}/mindspore/core/ops/logical_and.h
|
||||
${TOP_DIR}/mindspore/core/ops/logical_not.h
|
||||
${TOP_DIR}/mindspore/core/ops/logical_or.h
|
||||
${TOP_DIR}/mindspore/core/ops/lp_normalization.h
|
||||
${TOP_DIR}/mindspore/core/ops/lrn.h
|
||||
${TOP_DIR}/mindspore/core/ops/lsh_projection.h
|
||||
${TOP_DIR}/mindspore/core/ops/lstm.h
|
||||
${TOP_DIR}/mindspore/core/ops/maximum.h
|
||||
${TOP_DIR}/mindspore/core/ops/switch_layer.h
|
||||
${TOP_DIR}/mindspore/core/ops/mfcc.h
|
||||
${TOP_DIR}/mindspore/core/ops/minimum.h
|
||||
${TOP_DIR}/mindspore/core/ops/mod.h
|
||||
${TOP_DIR}/mindspore/core/ops/neg.h
|
||||
${TOP_DIR}/mindspore/core/ops/not_equal.h
|
||||
${TOP_DIR}/mindspore/core/ops/non_max_suppression.h
|
||||
${TOP_DIR}/mindspore/core/ops/one_hot.h
|
||||
${TOP_DIR}/mindspore/core/ops/ones_like.h
|
||||
${TOP_DIR}/mindspore/core/ops/prior_box.h
|
||||
${TOP_DIR}/mindspore/core/ops/quant_dtype_cast.h
|
||||
${TOP_DIR}/mindspore/core/ops/rank.h
|
||||
${TOP_DIR}/mindspore/core/ops/range.h
|
||||
${TOP_DIR}/mindspore/core/ops/reciprocal.h
|
||||
${TOP_DIR}/mindspore/core/ops/real_div.h
|
||||
${TOP_DIR}/mindspore/core/ops/reshape.h
|
||||
${TOP_DIR}/mindspore/core/ops/resize.h
|
||||
${TOP_DIR}/mindspore/core/ops/reverse_sequence.h
|
||||
${TOP_DIR}/mindspore/core/ops/reverse_v2.h
|
||||
${TOP_DIR}/mindspore/core/ops/rfft.h
|
||||
${TOP_DIR}/mindspore/core/ops/roi_pooling.h
|
||||
${TOP_DIR}/mindspore/core/ops/round.h
|
||||
${TOP_DIR}/mindspore/core/ops/rsqrt.h
|
||||
${TOP_DIR}/mindspore/core/ops/scatter_nd.h
|
||||
${TOP_DIR}/mindspore/core/ops/sgd.h
|
||||
${TOP_DIR}/mindspore/core/ops/shape.h
|
||||
${TOP_DIR}/mindspore/core/ops/sigmoid_cross_entropy_with_logits.h
|
||||
${TOP_DIR}/mindspore/core/ops/sin.h
|
||||
${TOP_DIR}/mindspore/core/ops/skip_gram.h
|
||||
${TOP_DIR}/mindspore/core/ops/smooth_l1_loss.h
|
||||
${TOP_DIR}/mindspore/core/ops/softmax.h
|
||||
${TOP_DIR}/mindspore/core/ops/softmax_cross_entropy_with_logits.h
|
||||
${TOP_DIR}/mindspore/core/ops/space_to_batch.h
|
||||
${TOP_DIR}/mindspore/core/ops/space_to_batch_nd.h
|
||||
${TOP_DIR}/mindspore/core/ops/space_to_depth.h
|
||||
${TOP_DIR}/mindspore/core/ops/sparse_softmax_cross_entropy_with_logits.h
|
||||
${TOP_DIR}/mindspore/core/ops/sparse_to_dense.h
|
||||
${TOP_DIR}/mindspore/core/ops/split.h
|
||||
${TOP_DIR}/mindspore/core/ops/sqrt.h
|
||||
${TOP_DIR}/mindspore/core/ops/squeeze.h
|
||||
${TOP_DIR}/mindspore/core/ops/square.h
|
||||
${TOP_DIR}/mindspore/core/ops/squared_difference.h
|
||||
${TOP_DIR}/mindspore/core/ops/stack.h
|
||||
${TOP_DIR}/mindspore/core/ops/strided_slice.h
|
||||
${TOP_DIR}/mindspore/core/ops/switch.h
|
||||
${TOP_DIR}/mindspore/core/ops/tensor_list_from_tensor.h
|
||||
${TOP_DIR}/mindspore/core/ops/tensor_list_get_item.h
|
||||
${TOP_DIR}/mindspore/core/ops/tensor_list_reserve.h
|
||||
${TOP_DIR}/mindspore/core/ops/tensor_list_set_item.h
|
||||
${TOP_DIR}/mindspore/core/ops/tensor_list_stack.h
|
||||
${TOP_DIR}/mindspore/core/ops/transpose.h
|
||||
${TOP_DIR}/mindspore/core/ops/unique.h
|
||||
${TOP_DIR}/mindspore/core/ops/unsorted_segment_sum.h
|
||||
${TOP_DIR}/mindspore/core/ops/unsqueeze.h
|
||||
${TOP_DIR}/mindspore/core/ops/unstack.h
|
||||
${TOP_DIR}/mindspore/core/ops/where.h
|
||||
${TOP_DIR}/mindspore/core/ops/zeros_like.h
|
||||
${TOP_DIR}/mindspore/core/ops/select.h
|
||||
${TOP_DIR}/mindspore/core/ops/scatter_nd_update.h
|
||||
${TOP_DIR}/mindspore/core/ops/gru.h
|
||||
${TOP_DIR}/mindspore/core/ops/non_zero.h
|
||||
${TOP_DIR}/mindspore/core/ops/invert_permutation.h
|
||||
${TOP_DIR}/mindspore/core/ops/size.h
|
||||
${TOP_DIR}/mindspore/core/ops/random_standard_normal.h
|
||||
${TOP_DIR}/mindspore/core/ops/crop_and_resize.h
|
||||
${TOP_DIR}/mindspore/core/ops/erf.h
|
||||
${TOP_DIR}/mindspore/core/ops/is_finite.h
|
||||
${TOP_DIR}/mindspore/core/ops/lin_space.h
|
||||
${TOP_DIR}/mindspore/core/ops/uniform_real.h
|
||||
${TOP_DIR}/mindspore/core/ops/splice.h
|
||||
${TOP_DIR}/mindspore/core/ops/log_softmax.h
|
||||
${TOP_DIR}/mindspore/core/ops/call.h
|
||||
${TOP_DIR}/mindspore/core/ops/custom.h
|
||||
${TOP_DIR}/mindspore/core/ops/cumsum.h
|
||||
${TOP_DIR}/mindspore/core/ops/split_with_overlap.h
|
||||
${TOP_DIR}/mindspore/core/ops/ragged_range.h
|
||||
${TOP_DIR}/mindspore/core/ops/glu.h
|
||||
${TOP_DIR}/mindspore/core/ops/tensor_array.h
|
||||
${TOP_DIR}/mindspore/core/ops/tensor_array_read.h
|
||||
${TOP_DIR}/mindspore/core/ops/tensor_array_write.h
|
||||
${TOP_DIR}/mindspore/core/ops/affine.h
|
||||
${TOP_DIR}/mindspore/core/ops/all_gather.h
|
||||
${TOP_DIR}/mindspore/core/ops/reduce_scatter.h
|
||||
${TOP_DIR}/mindspore/core/ops/dynamic_quant.h
|
||||
${TOP_DIR}/mindspore/core/ops/random_normal.h
|
||||
${TOP_DIR}/mindspore/core/ops/nllloss.h
|
||||
${TOP_DIR}/mindspore/core/ops/op_name.h
|
||||
${TOP_DIR}/mindspore/core/ops/tuple_get_item.h
|
||||
${TOP_DIR}/mindspore/core/ops/add.h
|
||||
${TOP_DIR}/mindspore/core/ops/div.h
|
||||
${TOP_DIR}/mindspore/core/ops/mul.h
|
||||
${TOP_DIR}/mindspore/core/ops/tuple_get_item.h
|
||||
${TOP_DIR}/mindspore/core/ops/scale.h
|
||||
${TOP_DIR}/mindspore/core/ops/sub.h
|
||||
${TOP_DIR}/mindspore/core/ops/avg_pool.h
|
||||
${TOP_DIR}/mindspore/core/ops/exp.h
|
||||
${TOP_DIR}/mindspore/core/ops/conv2d_transpose.h
|
||||
${TOP_DIR}/mindspore/core/ops/conv2d.h
|
||||
${TOP_DIR}/mindspore/core/ops/pow.h
|
||||
${TOP_DIR}/mindspore/core/ops/topk.h
|
||||
${TOP_DIR}/mindspore/core/ops/reduce.h
|
||||
${TOP_DIR}/mindspore/core/ops/arg_max.h
|
||||
${TOP_DIR}/mindspore/core/ops/max_pool.h
|
||||
${TOP_DIR}/mindspore/core/ops/prelu.h
|
||||
${TOP_DIR}/mindspore/core/ops/tile.h
|
||||
${TOP_DIR}/mindspore/core/ops/make_tuple.h
|
||||
${TOP_DIR}/mindspore/core/ops/base_operator.h
|
||||
${TOP_DIR}/mindspore/core/ops/return.h
|
||||
${TOP_DIR}/mindspore/core/ops/pad.h
|
||||
DESTINATION ${CONVERTER_ROOT_DIR}/include/core/ops
|
||||
COMPONENT ${RUNTIME_COMPONENT_NAME}
|
||||
)
|
||||
install(FILES
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/activation.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/add_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/adder_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/arg_max_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/arg_min_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/avg_pool_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/conv2d_backprop_filter_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/conv2d_backprop_input_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/conv2d_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/conv2d_transpose_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/div_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/embedding_lookup_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/exp_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/full_connection.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/layer_norm_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/l2_normalize_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/mat_mul_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/max_pool_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/mul_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/pad_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/partial_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/pow_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/prelu_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/reduce_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/scale_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/slice_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/sub_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/tile_fusion.h
|
||||
${TOP_DIR}/mindspore/core/ops/fusion/topk_fusion.h
|
||||
DESTINATION ${CONVERTER_ROOT_DIR}/include/core/ops/fusion
|
||||
COMPONENT ${RUNTIME_COMPONENT_NAME}
|
||||
)
|
||||
endfunction()
|
||||
# full mode will also package the files of lite_cv mode.
|
||||
if(MSLITE_MINDDATA_IMPLEMENT STREQUAL "full")
|
||||
# full header files
|
||||
|
|
@ -287,8 +498,7 @@ if(PLATFORM_ARM64)
|
|||
COMPONENT ${RUNTIME_COMPONENT_NAME})
|
||||
install(FILES ${IR_HEADER} DESTINATION ${CONVERTER_ROOT_DIR}/include/core/ir
|
||||
COMPONENT ${RUNTIME_COMPONENT_NAME})
|
||||
install(DIRECTORY ${TOP_DIR}/mindspore/core/ops/ DESTINATION ${CONVERTER_ROOT_DIR}/include/core/ops
|
||||
COMPONENT ${RUNTIME_COMPONENT_NAME} FILES_MATCHING PATTERN "*.h")
|
||||
__install_white_list_ops()
|
||||
install(FILES ${UTILS_HEADER} DESTINATION ${CONVERTER_ROOT_DIR}/include/core/utils
|
||||
COMPONENT ${RUNTIME_COMPONENT_NAME})
|
||||
install(DIRECTORY ${TOP_DIR}/mindspore/lite/build/schema/
|
||||
|
|
@ -660,8 +870,7 @@ else()
|
|||
COMPONENT ${RUNTIME_COMPONENT_NAME})
|
||||
install(FILES ${IR_HEADER} DESTINATION ${CONVERTER_ROOT_DIR}/include/core/ir
|
||||
COMPONENT ${RUNTIME_COMPONENT_NAME})
|
||||
install(DIRECTORY ${TOP_DIR}/mindspore/core/ops/ DESTINATION ${CONVERTER_ROOT_DIR}/include/core/ops
|
||||
COMPONENT ${RUNTIME_COMPONENT_NAME} FILES_MATCHING PATTERN "*.h")
|
||||
__install_white_list_ops()
|
||||
install(FILES ${UTILS_HEADER} DESTINATION ${CONVERTER_ROOT_DIR}/include/core/utils
|
||||
COMPONENT ${RUNTIME_COMPONENT_NAME})
|
||||
install(DIRECTORY ${TOP_DIR}/mindspore/lite/build/schema/ DESTINATION ${CONVERTER_ROOT_DIR}/include/schema
|
||||
|
|
|
|||
Loading…
Reference in New Issue