본문 바로가기

dev_AI_framework

계산 그래프 구성을 위한 반환 값 수정, root_node_list, leaf_node_list

이전까지 방법은 특정 레이어의 계산 그래프 생성 후, 다른 레이어들과 계산 그래프 연결을 통해 전체 계산 그래프 구성,

 

문제 발생 시점 : 특정 activaton_layer 의 경우, 특정 리프 노드가 다른 layer 의 루트 노드와 연결되지 않을 수 있음 ( operation : const ) 

 

layer 의 계산 그래프 반환에서 루트 노드가 아닌, 루트 노드와 다른 계산 그래프와 연결될 리프 노드, 두 개의 node_list 를 반환하도록 수정