def step_right_retains

bendtorch-core@0.1.0.0 · src/tensor/elementwise_PROOF.bend:262

import bendtorch-core@0.1.0.0/src/tensor/elementwise_PROOF.bend as ElementwisePROOF
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/core.bend as Core
import bendtorch-core@0.1.0.0/src/tensor/elementwise.bend as Elementwise
import bendtorch-core@0.1.0.0/src/tensor/elementwise_totality.bend as Totality

262def step_right_retains(+operation: Elementwise.Operation, -shape: Shape.Shape, +index: U32,
263  -original_left: Core.Dynamic, -original_right: Core.Dynamic, left: Core.Dynamic,
264  output: Core.Tensor<shape>, left_value: F32, pair: Core.Dynamic & Result<&2, &2, Core.Error, F32>,
265  -left_same: Totality.SameSource(original_left, left),
266  -right_same: Totality.SameSource(original_right, dynamic_read_owner(pair))) ->
267  StepSources<shape, original_left, original_right, Elementwise.step_right(operation, shape, index, left, output, left_value, pair)>:
268  (right, result) = pair
269  match result:
270    case Fail{error}:
271      StepSources{left_same, right_same}
272    case Done{value}:
273      step_write_retains(shape, original_left, original_right, left, right,
274        Core.write_flat(shape, output, index, Elementwise.scalar(operation, left_value, value)), left_same, right_same)

Uses (17)

Used by (1)