def folded_valid

stelliferous@0.0.2.0 · tensor_layout.bend:437

import Base
import stelliferous@0.0.2.0/storage_buffer.bend as Storage
import stelliferous@0.0.2.0/tensor.bend as Tensors

437def folded_valid(~Element: Data,name: String,valid: Bool,output: Storage.Buffer<Element>,shape: List<&2,U32>,result_shape: List<&2,U32>,fill: Element,
438  workers: U32) -> Tensors.Tensor<Element>:
439  match valid:
440    case True{}: Tensors.Tensor{output,result_shape,fill,Tensors.Usable{},workers}
441    case False{}: Tensors.Tensor{output,result_shape,fill,Tensors.Failed{Tensors.Problem{name,Tensors.InvalidArgument{shape}}},workers}