classify_dynamic_lno_onnx_expanded_nop Function

public function classify_dynamic_lno_onnx_expanded_nop(prefix, nodes, num_nodes) result(is_dynamic_lno)

Return true when the expanded-ONNX node cluster for prefix is a dynamic LNO.

Arguments

Type IntentOptional Attributes Name
character(len=*), intent(in) :: prefix

Expanded-ONNX layer prefix (e.g. "layer1")
type(onnx_node_type), intent(in) :: nodes(:)

Parsed ONNX nodes
integer, intent(in) :: num_nodes

Number of valid node entries

Return Value logical

Return value


Source Code

  function classify_dynamic_lno_onnx_expanded_nop(prefix, nodes, &
       num_nodes) result(is_dynamic_lno)
    !! Return true when the expanded-ONNX node cluster
    !! for prefix is a dynamic LNO.
    implicit none

    ! Arguments
    character(*), intent(in) :: prefix
    !! Expanded-ONNX layer prefix (e.g. "layer1")
    type(onnx_node_type), intent(in) :: nodes(:)
    !! Parsed ONNX nodes
    integer, intent(in) :: num_nodes
    !! Number of valid node entries

    logical :: is_dynamic_lno
    !! Return value

    is_dynamic_lno = &
         find_onnx_expanded_node_by_suffix( &
              nodes, num_nodes, prefix, 'Exp') .gt. 0 &
         .and. &
         find_onnx_expanded_node_by_suffix( &
              nodes, num_nodes, prefix, 'Exp_1') .gt. 0

  end function classify_dynamic_lno_onnx_expanded_nop