• 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
hardmax.prototxt 812 Bytes
ir_version: 3
producer_name: "nGraph ONNX Importer"
graph {
  node {
    input: "x"
    output: "y"
    op_type: "Hardmax"
      attribute {
      name: "axis"
      i: 2
      type: INT
    }
  }
  name: "hardmax_graph"
  input {
    name: "x"
    type {
      tensor_type {
        elem_type: 1
        shape {
          dim {
            dim_value: 3
          }
          dim {
            dim_value: 4
          }
          dim {
            dim_value: 5
          }
        }
      }
    }
  }
  output {
    name: "y"
    type {
      tensor_type {
        elem_type: 1
        shape {
          dim {
            dim_value: 3
          }
          dim {
            dim_value: 4
          }
          dim {
            dim_value: 5
          }
        }
      }
    }
  }
}
opset_import {
  version: 9
}