MCPcopy Create free account

hub / github.com/AlexanderLutsenko/nobuco / functions

Functions654 in github.com/AlexanderLutsenko/nobuco

Method__exit__
(self, *args)
nobuco/trace/trace.py:183
Method__exit__
(self, *args)
nobuco/trace/trace.py:198
Method__exit__
(self, *args)
nobuco/trace/trace.py:212
Method__init__
(self, unused_inputs, unreached_outputs, unused_nodes, unprovided_inputs)
nobuco/commons.py:43
Method__init__
(self, layer, backward_layer)
nobuco/node_converters/recurrent.py:14
Method__init__
(self, prefix, total=None, bar_format=None)
nobuco/vis/progress.py:7
Method__init__
(self, original_node)
nobuco/layers/stub.py:3
Method__init__
(self, original_node)
nobuco/layers/stub.py:14
Method__init__
(self, weight_shape, weight_dtype, trainable, *args, **kwargs)
nobuco/layers/weight.py:9
Method__init__
(self, func, channel_ordering_strategy, output_types=None, autocast=False)
nobuco/layers/channel_order.py:10
Method__init__
(self, op_descr_list, input_names, output_names, outputs_template, constants_dict=None, disconnected_tensors_d
nobuco/layers/container.py:8
Method__init__
(self, convert_func, validate_func, channel_ordering_strategy, autocast, reusable)
nobuco/converters/node_converter.py:11
Method__init__
(self, diff_abs: float, diff_rel: float, status: ValidationStatus)
nobuco/converters/validation.py:25
Method__init__
(self, converted_manually, is_implemented=True, is_duplicate=False, connectivity_status=None, converter=None)
nobuco/converters/validation.py:43
Method__init__
(self)
nobuco/addons/torch/dense_image_warp.py:13
Method__init__
(self, block_size)
nobuco/addons/torch/space_to_depth.py:13
Method__init__
(self, block_size)
nobuco/addons/torch/depth_to_space.py:13
Method__init__
(self)
nobuco/trace/tensor_storage.py:7
Method__init__
(self, enable_torch_tracing)
nobuco/trace/trace.py:159
Method__init__
(self, trace_shape)
nobuco/trace/trace.py:176
Method__init__
(self, is_allowed)
nobuco/trace/trace.py:191
Method__init__
(self)
nobuco/trace/trace.py:202
Method__init__
(self, parent)
nobuco/trace/trace.py:206
Method__init__
(self, op)
nobuco/entity/pytorch.py:17
Method__init__
(self, args, kwargs)
nobuco/entity/pytorch.py:23
Method__init__
(self)
nobuco/entity/pytorch.py:29
Method__init__
(self, style: str, is_last: bool, is_disconnected: bool, has_parent_outputs: bool)
nobuco/entity/pytorch.py:53
Method__init__
(self, wrapped_op: WrappedOp, module_name, parent_list, instance, input_args, input_kwargs, outputs, is_inplac
nobuco/entity/pytorch.py:61
Method__init__
(self, node: PytorchNode, children: Collection)
nobuco/entity/pytorch.py:116
Method__init__
(self, keras_op, pytorch_node: PytorchNode, validation_result: ValidationResult, conversion_r
nobuco/entity/keras.py:8
Method__init__
(self)
examples/dynamic_shape.py:17
Method__init__
(self)
examples/complex.py:17
Method__init__
(self)
examples/create_tensor.py:17
Method__init__
(self)
examples/conv_depthwise_tfjs.py:17
Method__init__
(self)
examples/conv_grouped.py:17
Method__init__
(self)
examples/control_for.py:17
Method__init__
(self)
examples/norm_example.py:19
Method__init__
(self)
examples/fourier.py:18
Method__init__
(self, crop_height, crop_width)
examples/static_crop.py:29
Method__init__
(self, iou_threshold)
examples/nms.py:18
Method__init__
(self)
examples/split.py:17
Method__init__
(self)
examples/slice_assign_mask.py:17
Method__init__
(self, embed_dim=128, num_heads=8)
examples/multihead-attention.py:18
Method__init__
(self)
examples/module_deepcopy.py:22
Method__init__
(self)
examples/dense_image_warp2.py:18
Method__init__
(self, linear_in, linear_out, alpha, *args, **kwargs)
examples/stateful1.py:34
Method__init__
(self)
examples/grad.py:17
Method__init__
(self, conv_pre, conv_true, conv_false, conv_shared, *args, **kwargs)
examples/control_if.py:41
Method__init__
(self)
examples/pixelshuffle.py:34
Method__init__
(self)
examples/channel_ordering.py:17
Method__init__
(self)
examples/efficient_reshape.py:17
Method__init__
(self)
examples/repeat.py:17
Method__init__
(self)
examples/control_while.py:17
Method__init__
(self)
examples/fusible.py:32
Method__init__
(self)
examples/reduce.py:18
Method__init__
(self)
examples/projection.py:18
Method__init__
(self)
examples/rnn.py:17
Method__init__
(self, forward_stateless, *args, **kwargs)
examples/stateful2.py:38
Method__init__
(self)
examples/grid_samplers.py:18
Method__init__
(self)
examples/converter_inside_converter.py:32
Method__init__
(self)
examples/control_if2.py:17
Method__init__
(self)
examples/stateful3.py:31
Method__init__
(self, shape, alpha, *args, **kwargs)
examples/stateful3.py:45
Method__init__
(self)
examples/paddings.py:18
Method__init__
(self)
examples/slice_assign.py:17
Method__init__
(self)
examples/math_ops.py:17
Method__init__
Simple block wrapping Mamba block with normalization and residual connection.
examples/mamba_minimal/model.py:146
Method__init__
A single Mamba block, as described in Figure 3 in Section 3.4 in the Mamba paper [1].
examples/mamba_minimal/model.py:179
Method__init__
(self, d_model: int, eps: float = 1e-5)
examples/mamba_minimal/model.py:335
Method__post_init__
(self)
examples/mamba_minimal/model.py:49
Method__str__
(self)
nobuco/layers/stub.py:20
Method__str__
(self)
nobuco/layers/channel_order.py:79
Method__str__
(self, tier=0, tier_statuses=None, validation_result_dict=None, conversion_res
nobuco/entity/pytorch.py:126
Method__torch_function__
(cls, func_raw, types, args=(), kwargs=None)
nobuco/trace/trace.py:23
Methodcall
(self, *args, **kwargs)
nobuco/layers/weight.py:31
Methodcall
(self, x, n)
examples/control_for.py:56
Methodcall
(self, q, k, v)
examples/multihead-attention.py:59
Methodcall
(self, x)
examples/stateful1.py:53
Methodcall
(self, x)
examples/control_if.py:59
Methodcall
(self, x)
examples/control_while.py:62
Methodcall
(self, x)
examples/stateful2.py:52
Methodcall
(self, inputs)
examples/control_if2.py:68
Methodcall
(self, x)
examples/stateful3.py:61
Methodcall
(self, l, x, deltaA, deltaB_u, C)
examples/mamba_minimal/convert.py:41
Functioncollect_func
(obj)
nobuco/convert.py:227
Functioncollect_func
(obj)
nobuco/layers/channel_order.py:85
Functioncollect_func
(obj)
nobuco/converters/type_cast.py:73
Functioncollect_func
(obj)
nobuco/converters/channel_ordering.py:37
Functioncollect_func
(obj)
nobuco/converters/scalars_to_tensors.py:11
Methodconnectivity_status_to_style
(self, tensor, self_status: ConnectivityStatus, parent_status: ConnectivityStatus, is_input)
nobuco/vis/console_stylizer.py:30
Functionconverter
(*ops, validate_func=validate_diff_default, channel_ordering_strategy=ChannelOrder
nobuco/converters/node_converter.py:30
FunctionconverterControlIf
(self, x)
examples/control_if.py:76
Functionconverter_BatchNorm
(self, input: Tensor)
nobuco/node_converters/normalization.py:17
Functionconverter_Conv1d
(self, input: Tensor)
nobuco/node_converters/convolution.py:17
Functionconverter_Conv2d
(self, input: Tensor)
nobuco/node_converters/convolution.py:301
Functionconverter_ConvTranspose1d
(self, input: Tensor, output_size: Optional[List[int]] = None)
nobuco/node_converters/convolution.py:167
Functionconverter_ConvTranspose2d
(self, input: Tensor, output_size: Optional[List[int]] = None)
nobuco/node_converters/convolution.py:458
Functionconverter_DetrFrozenBatchNorm
(self, input)
examples/detr.py:22
Functionconverter_Dropout
(self, input: Tensor)
nobuco/node_converters/dropout.py:12
Functionconverter_EMA
(self, x)
examples/stateful3.py:71
← previousnext →201–300 of 654, ranked by callers