fovi.arch package

KNN layers and shared neural-network primitives.

Complete architectures live in fovi.models.

Modules

• fovi.arch.knn
  • KNNBaseLayer
    • KNNBaseLayer._set_knn_padding_metadata()
    • KNNBaseLayer._mask_invalid_outputs()
    • KNNBaseLayer._as_geodesic_vertices()
    • KNNBaseLayer._compute_knns()
    • KNNBaseLayer._compute_geodesic_distances()
    • KNNBaseLayer._compute_euclidean_distances_batched()
    • KNNBaseLayer._compute_all_distances()
  • KNNGetterLayer
    • KNNGetterLayer.__init__()
    • KNNGetterLayer.forward()
  • KNNPoolingLayer
    • KNNPoolingLayer.__init__()
    • KNNPoolingLayer._compute_gaussian_weights()
    • KNNPoolingLayer.forward()
  • KNNConvLayer
    • KNNConvLayer.__init__()
    • KNNConvLayer._init_conv_like()
    • KNNConvLayer._pad_and_gather_knns()
    • KNNConvLayer._apply_local_rf()
    • KNNConvLayer._apply_local_rf_to_weights()
    • KNNConvLayer.forward()
    • KNNConvLayer.clear_optimized_cache()
    • KNNConvLayer.compute_reference_coords()
    • KNNConvLayer.compute_local_rf()
    • KNNConvLayer.load_conv2d_weights()
    • KNNConvLayer.load_conv3d_weights()
  • KNNDepthwiseSeparableConvLayer
    • KNNDepthwiseSeparableConvLayer.__init__()
    • KNNDepthwiseSeparableConvLayer._init_depthwise_separable_conv_like()
    • KNNDepthwiseSeparableConvLayer.forward()
  • KNNDepthwiseConvLayer
    • KNNDepthwiseConvLayer.__init__()
    • KNNDepthwiseConvLayer._init_depthwise_conv_like()
    • KNNDepthwiseConvLayer.forward()
  • compute_receptive_field()
  • compute_binary_receptive_field()
  • get_in_out_coords()
  • get_knn_conv_layer()
• fovi.arch.knn_autograd
  • dataclass()
  • TrainingMeta
    • TrainingMeta.cin
    • TrainingMeta.cout
    • TrainingMeta.nin
    • TrainingMeta.nout
    • TrainingMeta.k
    • TrainingMeta.v
    • TrainingMeta.p
    • TrainingMeta.p64
    • TrainingMeta.q
    • TrainingMeta.input_linear
    • TrainingMeta.weight_linear
    • TrainingMeta.input_linear_flat
    • TrainingMeta.weight_linear_flat
    • TrainingMeta.rev_rowptr
    • TrainingMeta.rev_col
    • TrainingMeta.device
    • TrainingMeta.__init__()
  • ensure_training_metadata()
  • _nout_chunk()
  • _pad_flat_input()
  • _effective_weight_t()
  • CompactTorchOps
    • CompactTorchOps.name
    • CompactTorchOps.forward()
    • CompactTorchOps.grad_input()
    • CompactTorchOps.grad_weight()
  • register_ops()
  • KNNConvFunction
    • KNNConvFunction.forward()
    • KNNConvFunction.backward()
  • compact_forward()
  • scatter_forward()
• fovi.arch.knn_cuda
  • forward()
  • grad_input()
  • grad_weight()
  • backward_combined()
  • CudaOps
    • CudaOps.name
    • CudaOps.forward()
    • CudaOps.grad_input()
    • CudaOps.grad_weight()
    • CudaOps.backward_combined()
  • KernelConfig
    • KernelConfig.async_copy
    • KernelConfig.bk
    • KernelConfig.bm
    • KernelConfig.bn
    • KernelConfig.ksteps
    • KernelConfig.rb
    • KernelConfig.vec
    • KernelConfig.wm
    • KernelConfig.wn
  • default_config()
  • default_grad_input_config()
  • default_grad_weight_config()
  • default_grad_weight_ksplit()
  • clear_kernel_cache()
• fovi.arch.knn_gather_gemm
  • _node_index()
  • _scatter_matrix()
  • GatherGemmOps
    • GatherGemmOps.name
    • GatherGemmOps.forward()
    • GatherGemmOps.grad_input()
    • GatherGemmOps.grad_weight()
  • _register_ops()
• fovi.arch.knn_optimization
  • _is_gather_gemm_layer()
  • _native_cuda_supported()
  • clear_cache()
  • select_backend()
  • optimized_forward()
• fovi.arch.knn_pool_cuda
  • PoolMeta
    • PoolMeta.nin
    • PoolMeta.nout
    • PoolMeta.k
    • PoolMeta.indices
    • PoolMeta.indices_i64
    • PoolMeta.count
    • PoolMeta.inv_count
    • PoolMeta.rev_rowptr
    • PoolMeta.rev_kn
    • PoolMeta.device
    • PoolMeta.__init__()
  • pool_meta_from_indices()
  • ensure_pool_metadata()
  • pool_forward()
  • pool_backward()
  • KNNPoolFunction
    • KNNPoolFunction.forward()
    • KNNPoolFunction.backward()
  • pool_function()
  • optimized_pool_forward()
  • clear_pool_cache()
• fovi.arch.knn_warp
  • _make_uncached_r2_kernel()
  • _make_uncached_r4_kernel()
  • _canary_launch_ok()
  • _get_uncached_batch_config()
  • _select_uncached_batch_config()
  • _pad_bias_to_tiles()
  • run_cached()
  • _make_grad_input_stage2_kernel()
  • _make_grad_weight_stage3_kernel()
  • _make_grad_weight_stage4_kernel()
  • _make_grad_weight_stage2_kernel()
  • _canary_grad_stage_ok()
  • _get_grad_kernel()
  • _pad_flat_input_fp16()
  • grad_input()
  • grad_weight()
  • WarpCompactOps
    • WarpCompactOps.name
    • WarpCompactOps.forward()
    • WarpCompactOps.grad_input()
    • WarpCompactOps.grad_weight()
  • WarpTrainOps
    • WarpTrainOps.name
    • WarpTrainOps.forward()
    • WarpTrainOps.grad_input()
    • WarpTrainOps.grad_weight()
  • _register_ops()
  • run_uncached()
• fovi.arch.mlp
  • LayerBlock
    • LayerBlock.__init__()
  • get_mlp()
  • MLPWrapper
    • MLPWrapper.__init__()
    • MLPWrapper.forward()
• fovi.arch.norm
  • KNNChannelNorm
    • KNNChannelNorm.__init__()
    • KNNChannelNorm.forward()
  • KNNBatchNorm
    • KNNBatchNorm.__init__()
  • KNNCoordNorm
    • KNNCoordNorm.__init__()
  • KNNLayerNorm
    • KNNLayerNorm.__init__()
  • KNNGroupNorm
    • KNNGroupNorm.__init__()
  • find_valid_num_groups()
  • get_norm()
• fovi.arch.polar
  • PolarPadder
    • PolarPadder.__init__()
    • PolarPadder.forward()
  • convert_model_to_polar()
• fovi.arch.wrapper
  • BackboneProjectorWrapper
    • BackboneProjectorWrapper.__init__()
    • BackboneProjectorWrapper.forward()