def clone_result

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

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/storage/main.bend as Storage
import bendtorch-core@0.1.0.0/src/tensor/lifecycle.bend as Lifecycle

324def clone_result(-shape: Shape.Shape, -original: Storage.Storage,
325  -valid: {Lifecycle.logical_length(original) == Shape.logical_numel(Shape.dimensions(shape)) : Nat},
326  pair: Storage.Storage & Storage.Storage,
327  -same: {pair == Storage.clone(original) : Storage.Storage & Storage.Storage}) -> Tensor<shape> & Tensor<shape>:
328  (first, second) = pair
329  (TTensor{first, Equal.trans(Nat, Lifecycle.logical_length(first), Lifecycle.logical_length(original),
330    Shape.logical_numel(Shape.dimensions(shape)),
331    %Equal.sym(Storage.Storage & Storage.Storage, (first, second), Storage.clone(original), same) :
332      {Lifecycle.first_length(_) == Lifecycle.logical_length(original) : Nat}
333    Lifecycle.clone_first_length(original), valid)},
334  TTensor{second, Equal.trans(Nat, Lifecycle.logical_length(second), Lifecycle.logical_length(original),
335    Shape.logical_numel(Shape.dimensions(shape)),
336    %Equal.sym(Storage.Storage & Storage.Storage, (first, second), Storage.clone(original), same) :
337      {Lifecycle.second_length(_) == Lifecycle.logical_length(original) : Nat}
338    Lifecycle.clone_second_length(original), valid)})