Skip to main content

Browse the archive

Revision ddd614adc93aa4e75057bb1226a3e2adf797993e authored by Steven Johnson on 22 December 2020, 23:53:35 UTC, committed by Steven Johnson on 22 December 2020, 23:53:35 UTC 
TFLiteParser should handle the 2-input form of Reshape
 
The new shape can be specified in a 2nd input (instead of the ReshapeParams); we should handle this properly
1 parent 7916c84
Raw File
CMakeLists.txt 
if (NOT TARGET Halide::PyStubs)
    add_library(Halide_PyStubs STATIC ${CMAKE_CURRENT_LIST_DIR}/PyStubImpl.cpp)
    add_library(Halide::PyStubs ALIAS Halide_PyStubs)

    target_link_libraries(Halide_PyStubs PRIVATE pybind11::module)
    target_link_libraries(Halide_PyStubs PUBLIC Halide::Halide)
    set_target_properties(Halide_PyStubs PROPERTIES
                          EXPORT_NAME PyStubs
                          VISIBILITY_INLINES_HIDDEN TRUE
                          POSITION_INDEPENDENT_CODE ON)
endif ()

include(${CMAKE_CURRENT_LIST_DIR}/AddHalideGeneratorPython.cmake)
back to top