def apply_indexed_run

stelliferous@0.0.2.0 · tensor_access.bend:130

import stelliferous@0.0.2.0/tensor_access.bend as TensorAccess
import Base
import stelliferous@0.0.2.0/tensor_view.bend as Views
import stelliferous@0.0.2.0/traversal_array.bend as Traversal

130# Reads a contiguous view and writes a separate contiguous output from zero.
131def apply_indexed_run(~Element: Data,~operation: U32 -> Element -> Element,count: Nat,view: Views.View,
132  input: Array<Element>,output: Array<Element>) -> Array<Element> & Array<Element>:
133  Views.View{axes,offset} = view
134  Traversal.finish(~Array<Element>,~Element,~Traversal.Offsets,
135    Traversal.run(~Array<Element>,~Element,~Traversal.Offsets,~read_indexed(~Element,~operation),~Traversal.destination_offset,count,
136      Traversal.State{input,output,0,Traversal.Offsets{offset,0}}))

Uses (11)

Used by (3)