find_node_initialiser_index Function

public function find_node_initialiser_index(node, inits, num_inits)

Return the first initialiser referenced by a node's inputs.

Arguments

Type IntentOptional Attributes Name
type(onnx_node_type), intent(in) :: node

Parsed ONNX node whose inputs may reference an initialiser

type(onnx_initialiser_type), intent(in) :: inits(:)

Parsed ONNX initialisers

integer, intent(in) :: num_inits

Number of valid initialiser entries

Return Value integer


Source Code

  integer function find_node_initialiser_index(node, inits, num_inits)
    !! Return the first initialiser referenced by a node's inputs.
    implicit none

    ! Arguments
    type(onnx_node_type), intent(in) :: node
    !! Parsed ONNX node whose inputs may reference an initialiser
    type(onnx_initialiser_type), intent(in) :: inits(:)
    !! Parsed ONNX initialisers
    integer, intent(in) :: num_inits
    !! Number of valid initialiser entries

    ! Local variables
    integer :: i, init_idx
    !! Loop index and candidate initialiser index

    find_node_initialiser_index = 0
    if(.not.allocated(node%inputs)) return

    do i = 1, size(node%inputs)
       init_idx = find_initialiser_by_name(node%inputs(i), inits, num_inits)
       if(init_idx .gt. 0)then
          find_node_initialiser_index = init_idx
          return
       end if
    end do

  end function find_node_initialiser_index