• Tomasz Dołbniak's avatar
    [ONNX] Shrink op support (#3024) · 250dddbc
    Tomasz Dołbniak authored
    * Initial implementation of the Shrink op
    
    * Multiply the values by the correct masks
    
    * Basic test case for Shrink with floats
    
    * Shrink test on integers
    
    * Code formatting
    
    * Shrink documentation and typo fix
    
    * Rephrase the Shrink docs
    
    * Out of <memory> ;)
    250dddbc
shrink_float.prototxt 675 Bytes
ir_version: 3
producer_name: "nGraph ONNX Importer"
graph {
  node {
    input: "x"
    output: "y"
    op_type: "Shrink"
    attribute {
      name: "lambd"
      f: 1.5
      type: FLOAT
    }
    attribute {
      name: "bias"
      f: 0.5
      type: FLOAT
    }
  }
  name: "shrink_graph"
  input {
    name: "x"
    type {
      tensor_type {
        elem_type: 1
        shape {
          dim {
            dim_value: 11
          }
        }
      }
    }
  }
  output {
    name: "y"
    type {
      tensor_type {
        elem_type: 1
        shape {
          dim {
            dim_value: 11
          }
        }
      }
    }
  }
}
opset_import {
  version: 9
}