• Nishant Patel's avatar
    Simple bf16 test case for CPU backend (#3598) · 7ea34fc2
    Nishant Patel authored
    * test bfloat16 with CPU maxpool
    
    * Move test to cpu_test
    
    * convert result from bfloat16 to float
    
    * Modify result op to support bfloat16
    
    * Correct comment
    
    * Add a simple convolution test case
    
    * initialize floats with float literals
    
    * Guard bf16 tests with MKLDNN_VERSION_MAJOR >= 1
    7ea34fc2
Name
Last commit
Last update
..
autodiff Loading commit data...
CMakeLists.txt Loading commit data...
all_close.hpp Loading commit data...
all_close_f.cpp Loading commit data...
all_close_f.hpp Loading commit data...
float_util.cpp Loading commit data...
float_util.hpp Loading commit data...
known_element_types.hpp Loading commit data...
matcher.hpp Loading commit data...
ndarray.hpp Loading commit data...
random.hpp Loading commit data...
test_case.cpp Loading commit data...
test_case.hpp Loading commit data...
test_control.cpp Loading commit data...
test_control.hpp Loading commit data...
test_tools.cpp Loading commit data...
test_tools.hpp Loading commit data...
type_prop.hpp Loading commit data...