def depth

stelliferous@0.0.2.0 · tensor_convolution.bend:66

import Base
import stelliferous@0.0.2.0/kernel_pipeline.bend as Pipeline
import stelliferous@0.0.2.0/default_kernel_cost.bend as DefaultCost
import stelliferous@0.0.2.0/math_activation.bend as Activations

66# The input tensor's workers bound the split, as a thread pool bounds PyTorch's parallel_for.
67def depth(work: Pipeline.Work,input_capacity: U32,bias_capacity: U32,workers: U32,+activation: Activations.Activation) -> U32:
68  configured_depth(work,input_capacity,bias_capacity,workers,activation,DefaultCost.configuration())