• Ted Themistokleous's avatar
    Initial changes for jit gather implementation · 86ff3faf
    Ted Themistokleous authored
    Taken from gatherND.cpp and modified so we include the axis parameter as opposed
    to the batch_dims attribute.
    
    Should always exist since we default this to zero when no axis is provided from
    the instruction
    
    Work in progress for the .hpp jit side.
    86ff3faf
gather.cpp 3.09 KB