~/jendhub

def fitted_same

bendtorch-core@0.1.0.0 · src/autograd/reduce_to_fold_PROOF.bend:144

import bendtorch-core@0.1.0.0/src/autograd/reduce_to_fold_PROOF.bend as ReduceToFoldPROOF
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/layout.bend as Layout
import bendtorch-core@0.1.0.0/src/tensor/permutation_mapping.bend as Mapping

144def fitted_same(+shape: Shape.Shape, +plan: Shape.BroadcastPlan,
145  same: {Shape.dimensions(Shape.output_shape(plan)) == Shape.dimensions(shape) : List<&2, U32>}) ->
146  {Layout.words_same(Shape.dimensions(Shape.output_shape(plan)), Shape.dimensions(shape)) == True{} : Bool}:
147  %Equal.sym(List<&2, U32>, Shape.dimensions(Shape.output_shape(plan)), Shape.dimensions(shape), same) :
148    {Layout.words_same(_, Shape.dimensions(shape)) == True{} : Bool}
149  Mapping.words_same_refl(Shape.dimensions(shape))

Uses (10)

Used by (2)