IRON 1a5eed49d3c0721a318ac369f725acc96b7c4584
Loading...
Searching...
No Matches
Namespaces | Functions
conv.py File Reference

Namespaces

namespace  iron
 
namespace  iron.kernels
 
namespace  iron.kernels.conv
 

Functions

list iron.kernels.conv._i32s (int n)
 
ExternalFunction iron.kernels.conv.conv2dk1 (int input_width=32, int input_channels=64, int output_channels=64, act_dtype=np.int8)
 
ExternalFunction iron.kernels.conv.conv2dk3 (int input_width=32, int input_channels=64, int output_channels=64, act_dtype=np.int8, int|None weight_output_channels=None)
 
ExternalFunction iron.kernels.conv.conv2dk1_skip (int input_width=32, int input_channels=64, int output_channels=64, act_dtype=np.int8)
 
ExternalFunction iron.kernels.conv.conv2dk1_i8 (int input_width=32, int input_channels=64, int output_channels=64)
 
ExternalFunction iron.kernels.conv.conv2dk14 (int input_width=224, int input_channels=16, int output_channels=16, int kernel_width=14)
 
ExternalFunction iron.kernels.conv.conv2dk1_skip_init (int input_width=32, int input_channels=64, int output_channels=64, act_dtype=np.int8, int|None skip_input_channels=None)
 
ExternalFunction iron.kernels.conv.bn_conv2dk1_relu (int input_width=32, int input_channels=64, int output_channels=64)
 
ExternalFunction iron.kernels.conv.bn_conv2dk3 (int input_width=32, int input_channels=64, int output_channels=64)
 
ExternalFunction iron.kernels.conv.bn_conv2dk1_i8 (int input_width=32, int input_channels=64, int output_channels=64)
 
ExternalFunction iron.kernels.conv.bn_conv2dk1_skip (int input_width=32, int input_channels=64, int output_channels=64, skip_dtype=np.uint8)
 
ExternalFunction iron.kernels.conv.bn_conv2dk3_dw (int input_width=32, int input_channels=64, int output_channels=64, int stride=1)
 
ExternalFunction iron.kernels.conv.bn_conv2dk1_relu_xy_pool_padded (int input_width=7, int input_channels=80, int output_channels=1280, int|None weight_chunk_count=None)
 
None iron.kernels.conv._validate_bn_block_index (int block_index, str factory_name)
 
ExternalFunction iron.kernels.conv.bn_conv2dk1_partial_put_i8 (int input_width=7, int input_channels=80, int weight_count=4800, *int block_index=13)
 
ExternalFunction iron.kernels.conv.bn_conv2dk1_partial_get_relu_i8 (int input_width=7, int input_channels=80, int output_channels=480, int weight_count=4800, *int block_index=13)
 
ExternalFunction iron.kernels.conv.bn_conv2dk3_dw_out_split (int input_width=7, int input_channels=480, int output_split_channels=240, *int block_index=13)
 
ExternalFunction iron.kernels.conv.bn_conv2dk1_input_split_partial_put_ui8 (int input_width=7, int input_channels=240, int weight_count=9600, *int block_index=13)
 
ExternalFunction iron.kernels.conv.bn_conv2dk1_input_split_partial_skip_get (int input_width=7, int input_channels=240, int output_channels=80, int weight_count=9600, *int block_index=13)
 
ExternalFunction iron.kernels.conv.bn_fc_relu_ui16_pad (int input_channels=1280, int output_channels=16, int|None weight_chunk_count=None)