Class: Kumi::IR::DF::Builder

Inherits:
Base::Builder show all
Defined in:
lib/kumi/ir/df.rb

Instance Attribute Summary

Attributes inherited from Base::Builder

#current_block, #function, #ir_module

Instance Method Summary collapse

Methods inherited from Base::Builder

#emit, #initialize, #new_block, #set_block

Constructor Details

This class inherits a constructor from Kumi::IR::Base::Builder

Instance Method Details

#array_build(result:, elements:, axes:, dtype:, metadata: {}) ⇒ Object 



58
59
60
 
# File 'lib/kumi/ir/df.rb', line 58

def array_build(result:, elements:, axes:, dtype:, metadata: {})
  append Ops::ArrayBuild.new(result:, elements:, axes:, dtype:, metadata:)
end

#array_get(result:, array:, index:, axes:, dtype:, oob:, metadata: {}) ⇒ Object 



62
63
64
 
# File 'lib/kumi/ir/df.rb', line 62

def array_get(result:, array:, index:, axes:, dtype:, oob:, metadata: {})
  append Ops::ArrayGet.new(result:, array:, index:, axes:, dtype:, oob:, metadata:)
end

#array_len(result:, array:, axes:, dtype:, metadata: {}) ⇒ Object 



66
67
68
 
# File 'lib/kumi/ir/df.rb', line 66

def array_len(result:, array:, axes:, dtype:, metadata: {})
  append Ops::ArrayLen.new(result:, array:, axes:, dtype:, metadata:)
end

#axis_broadcast(result:, value:, from_axes:, to_axes:, dtype:, metadata: {}) ⇒ Object 



82
83
84
 
# File 'lib/kumi/ir/df.rb', line 82

def axis_broadcast(result:, value:, from_axes:, to_axes:, dtype:, metadata: {})
  append Ops::AxisBroadcast.new(result:, value:, from_axes:, to_axes:, axes: to_axes, dtype:, metadata:)
end

#axis_index(result:, axis:, axes:, dtype:, metadata: {}) ⇒ Object 



70
71
72
 
# File 'lib/kumi/ir/df.rb', line 70

def axis_index(result:, axis:, axes:, dtype:, metadata: {})
  append Ops::AxisIndex.new(result:, axis:, axes:, dtype:, metadata:)
end

#axis_shift(result:, source:, axis:, offset:, policy:, axes:, dtype:, metadata: {}) ⇒ Object 



78
79
80
 
# File 'lib/kumi/ir/df.rb', line 78

def axis_shift(result:, source:, axis:, offset:, policy:, axes:, dtype:, metadata: {})
  append Ops::AxisShift.new(result:, source:, axis:, offset:, policy:, axes:, dtype:, metadata:)
end

#constant(result:, value:, axes:, dtype:, metadata: {}) ⇒ Object 



22
23
24
 
# File 'lib/kumi/ir/df.rb', line 22

def constant(result:, value:, axes:, dtype:, metadata: {})
  append Ops::Constant.new(result:, value:, axes:, dtype:, metadata:)
end

#decl_ref(result:, name:, axes:, dtype:, metadata: {}) ⇒ Object 



50
51
52
 
# File 'lib/kumi/ir/df.rb', line 50

def decl_ref(result:, name:, axes:, dtype:, metadata: {})
  append Ops::DeclRef.new(result:, name:, axes:, dtype:, metadata:)
end

#fold(result:, fn:, arg:, axes:, dtype:, metadata: {}) ⇒ Object 



46
47
48
 
# File 'lib/kumi/ir/df.rb', line 46

def fold(result:, fn:, arg:, axes:, dtype:, metadata: {})
  append Ops::Fold.new(result:, fn:, arg:, axes:, dtype:, metadata:)
end

#import_call(result:, fn_name:, source_module:, args:, mapping_keys:, axes:, dtype:, metadata: {}) ⇒ Object 



74
75
76
 
# File 'lib/kumi/ir/df.rb', line 74

def import_call(result:, fn_name:, source_module:, args:, mapping_keys:, axes:, dtype:, metadata: {})
  append Ops::ImportCall.new(result:, fn_name:, source_module:, args:, mapping_keys:, axes:, dtype:, metadata:)
end

#load_field(result:, object:, field:, axes:, dtype:, metadata: {}, plan_ref: nil) ⇒ Object 



30
31
32
 
# File 'lib/kumi/ir/df.rb', line 30

def load_field(result:, object:, field:, axes:, dtype:, metadata: {}, plan_ref: nil)
  append Ops::LoadField.new(result:, object:, field:, axes:, dtype:, metadata:, plan_ref:)
end

#load_input(result:, key:, axes:, dtype:, chain: [], metadata: {}, source_plan: nil, plan_ref: nil) ⇒ Object 



26
27
28
 
# File 'lib/kumi/ir/df.rb', line 26

def load_input(result:, key:, axes:, dtype:, chain: [], metadata: {}, source_plan: nil, plan_ref: nil)
  append Ops::LoadInput.new(result:, key:, axes:, dtype:, chain:, metadata:, source_plan:, plan_ref:)
end

#make_object(result:, inputs:, keys:, axes:, dtype:, metadata: {}) ⇒ Object 



54
55
56
 
# File 'lib/kumi/ir/df.rb', line 54

def make_object(result:, inputs:, keys:, axes:, dtype:, metadata: {})
  append Ops::MakeObject.new(result:, inputs:, keys:, axes:, dtype:, metadata:)
end

#map(result:, fn:, args:, axes:, dtype:, metadata: {}) ⇒ Object 



34
35
36
 
# File 'lib/kumi/ir/df.rb', line 34

def map(result:, fn:, args:, axes:, dtype:, metadata: {})
  append Ops::Map.new(result:, fn:, args:, axes:, dtype:, metadata:)
end

#reduce(result:, fn:, arg:, axes:, over_axes:, dtype:, metadata: {}) ⇒ Object 



42
43
44
 
# File 'lib/kumi/ir/df.rb', line 42

def reduce(result:, fn:, arg:, axes:, over_axes:, dtype:, metadata: {})
  append Ops::Reduce.new(result:, fn:, arg:, axes:, over_axes:, dtype:, metadata:)
end

#select(result:, cond:, on_true:, on_false:, axes:, dtype:, metadata: {}) ⇒ Object 



38
39
40
 
# File 'lib/kumi/ir/df.rb', line 38

def select(result:, cond:, on_true:, on_false:, axes:, dtype:, metadata: {})
  append Ops::Select.new(result:, cond:, on_true:, on_false:, axes:, dtype:, metadata:)
end