def combine_dense_with

stelliferous@0.0.2.0 · tensor.bend:327

import Base
import stelliferous@0.0.2.0/storage_buffer.bend as Storage
import stelliferous@0.0.2.0/tensor_view.bend as Views

327# Band layout is kept only for equal extents: operands that need broadcasting
328# are gathered and combined on dense storage.
329def combine_dense_with(~Element: Data,~kernel: @+count: Nat -> @+right_view: Views.View -> @+offset: U32 -> Storage.Buffer<Element> -> Storage.Buffer<Element> -> Storage.Buffer<Element> & Storage.Buffer<Element>,name: String,left: Tensor<Element>,right: Tensor<Element>) -> Tensor<Element>:
330  with_dense(~Element,~Tensor<Element>,left,left => left_shape => fill => left_status => left_workers =>
331    with_dense(~Element,~Tensor<Element>,right,right => right_shape => right_fill => right_status => right_workers =>
332      combine_status_with(~Element,~kernel,name,first_failure(left_status,right_status),left,left_shape,fill,right,right_shape,U32.max(left_workers,right_workers))))