• Tomasz Dołbniak's avatar
    [ONNX] Hardmax support in ONNX importer (#2869) · ae352fa4
    Tomasz Dołbniak authored
    * [ONNX] Hardmax implementation in the onnx importer
    
    * [ONNX] More generic handling of types in hardmax
    
    * [ONNX] Support for doubles in EmbeddingLookup CPU builder
    
    * [ONNX] Throw when the provided axis is out of range
    
    * [ONNX] Skip the hardmax test on GPU
    
    * Unused headers clean-up
    
    * refactor: move the identity matrix generator to common.hpp
    
    * ASSERT_VALID_ARGUMENT for axis range validation
    
    * Adapt to the code changes in master
    ae352fa4
Name
Last commit
Last update
..
mxnet Loading commit data...
onnx Loading commit data...
tensorflow Loading commit data...
CMakeLists.txt Loading commit data...
conv_bias.json Loading commit data...
tf_conv_mnist_nhwc.json Loading commit data...