Module: NDAV::OrtValue::MemoryViewable

Defined in:
lib/ndav/ort_value.rb

Instance Method Summary collapse

Instance Method Details

#ndav_descriptor ⇒ Object 



87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
 
# File 'lib/ndav/ort_value.rb', line 87

def ndav_descriptor(**)
  item_size = TYPE_SIZES[element_type]
  unless item_size
    warn "unsupported type: #{element_type}"
    return false
  end

  byte_size = shape.reduce(item_size, :*)

  {
    data: ::Fiddle::Pointer.new(data_ptr.address, byte_size),
    shape:,
    strides: ::NDAV.default_strides(shape:, item_size:),
    format: TYPE_TO_FORMAT[element_type],
    byte_size:,
    readonly?: true,
    sub_offsets: nil
  }
end