CMakeLists.txt 1.6 KB
Newer Older
H
hong19860320 已提交
1 2 3 4 5
if(NOT LITE_WITH_APU)
  return()
endif()


6
lite_cc_library(subgraph_bridge_utility_apu SRCS utility.cc DEPS tensor neuron_adapter)
H
hong19860320 已提交
7 8 9 10 11 12 13 14 15 16
lite_cc_library(subgraph_bridge_graph_apu SRCS graph.cc DEPS subgraph_bridge_utility_apu)

set(apu_subgraph_bridge_deps subgraph_bridge_registry subgraph_bridge_utility_apu subgraph_bridge_graph_apu)

lite_cc_library(subgraph_bridge_conv_op_apu SRCS conv_op.cc DEPS ${apu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_elementwise_ops_apu SRCS elementwise_ops.cc DEPS ${apu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_act_op_apu SRCS act_op.cc DEPS ${apu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_pool_op_apu SRCS pool_op.cc DEPS ${apu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_softmax_op_apu SRCS softmax_op.cc DEPS ${apu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_fc_op_apu SRCS fc_op.cc DEPS ${apu_subgraph_bridge_deps})
17 18
lite_cc_library(subgraph_bridge_concat_op_apu SRCS concat_op.cc DEPS ${apu_subgraph_bridge_deps})
lite_cc_library(subgraph_bridge_conv_transpose_op_apu SRCS conv_transpose_op.cc DEPS ${apu_subgraph_bridge_deps})
H
hong19860320 已提交
19 20 21 22 23 24 25 26 27 28 29


set(apu_subgraph_bridges
        subgraph_bridge_registry
        subgraph_bridge_utility_apu
        subgraph_bridge_conv_op_apu
        subgraph_bridge_elementwise_ops_apu
        subgraph_bridge_act_op_apu
        subgraph_bridge_softmax_op_apu
        subgraph_bridge_fc_op_apu
        subgraph_bridge_pool_op_apu
30 31
	subgraph_bridge_conv_transpose_op_apu
	subgraph_bridge_concat_op_apu
H
hong19860320 已提交
32 33 34
        CACHE INTERNAL "apu_subgraph_bridges")

message(STATUS "+++++ apu_subgraph_bridges: ${apu_subgraph_bridges}")