Module: NDAV::OrtValue::MemoryViewable

Defined in:
lib/ndav/ort_value.rb

Instance Method Summary collapse

Instance Method Details

#ndav_descriptor ⇒ Object 



112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
 
# File 'lib/ndav/ort_value.rb', line 112

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