def cell_of

bendtorch-core@0.1.0.0 · src/tensor/elementwise_parallel_proof.bend:146

import bendtorch-core@0.1.0.0/src/tensor/elementwise_parallel_proof.bend as ElementwiseParallelProof
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/elementwise.bend as Elementwise

146# The elementwise cell function of `Parallel.build`, from the input arrays.
147def cell_of(+operation: Elementwise.Operation, +plan: Shape.BroadcastPlan, +position: Nat, A: Array<F32>, B: Array<F32>, inside: Bool) ->
148  F32 & (Array<F32> & Array<F32>):
149  Elementwise.cell_at(Elementwise.walk(Elementwise.Request{operation, plan}), position, (A, B), inside)

Uses (11)

Used by (3)