MCPcopy Create free account

hub / github.com/AlexanderLutsenko/nobuco / functions

Functions654 in github.com/AlexanderLutsenko/nobuco

⨍ Functions 654◇ Types & classes 95↳ Endpoints 3
↓ 52 callersMethodconvert
(self, *args, _pytorch_node: PytorchNode = None, **kwargs)
nobuco/converters/node_converter.py:18
↓ 31 callersFunctionget_channel_order
(tensor)
nobuco/converters/channel_ordering.py:16
↓ 24 callersFunctioncollect_recursively
(obj, classes)
nobuco/util.py:60
↓ 19 callersMethodget
(self, tensor)
nobuco/trace/tensor_storage.py:10
↓ 19 callersFunctionget_torch_tensor_identifier
(tensor)
nobuco/util.py:19
↓ 18 callersMethodstylize
(self, text: str, style: str)
nobuco/vis/console_stylizer.py:13
↓ 16 callersFunctiondim_pytorch2keras
(dim, num_dims)
nobuco/converters/tensor.py:16
↓ 16 callersFunctiondtype_pytorch2keras
(dtype)
nobuco/converters/type_cast.py:27
↓ 15 callersMethodupdate
(self, n=1)
nobuco/vis/progress.py:23
↓ 14 callersFunctionset_channel_order
(tensor, channel_order: ChannelOrder)
nobuco/converters/channel_ordering.py:11
↓ 12 callersFunctionreduce_func
(inner_func: Callable, n_dims: int)
nobuco/node_converters/reduce.py:21
↓ 11 callersFunctionperm_keras2pytorch
(n_dims)
nobuco/converters/tensor.py:48
↓ 11 callersFunctionreplace_recursively_func
(obj, collect_func: Callable[[object], bool], replace_func: Callable[[object], object])
nobuco/util.py:66
↓ 10 callersMethodop_undecorate
(callable: Callable)
nobuco/trace/trace.py:228
↓ 9 callersFunction_permute
(perm)
nobuco/converters/tensor.py:76
↓ 8 callersFunction_ensure_iterable
(iterable)
nobuco/converters/tensor.py:92
↓ 8 callersMethodget_type
(self)
nobuco/entity/pytorch.py:101
↓ 8 callersFunctiontype_func
(self, dtype=None, non_blocking=False, **kwargs)
nobuco/node_converters/tensor_cast.py:25
↓ 7 callersMethodfrom_pretrained
Load pretrained weights from HuggingFace into model. Args: pretrained_model_name: One of * 'state-spaces/mamba_mi
examples/mamba_minimal/model.py:98
↓ 7 callersMethodget_op
(self)
nobuco/entity/pytorch.py:108
↓ 7 callersMethodis_decorated
(callable)
nobuco/trace/trace.py:224
↓ 6 callersFunction_dim_make_positive
(dim, n_dims, add_one=False)
nobuco/converters/tensor.py:4
↓ 6 callersMethodfill
(self, hierarchy)
nobuco/entity/pytorch.py:33
↓ 6 callersFunctionpermute_pytorch2keras
(xs)
nobuco/converters/tensor.py:62
↓ 6 callersFunctionwrap_torch_tensors_recursively
(obj, annotate=True)
nobuco/trace/trace.py:52
↓ 5 callersFunction_permute_inner
(perm_original, allow_lazy=True)
nobuco/node_converters/tensor_manipulation.py:19
↓ 5 callersFunctiongather
(input, y, x, b, h, w, c, padding_mode)
nobuco/node_converters/grid_sampling.py:13
↓ 5 callersFunctionperm_pytorch2keras
(n_dims)
nobuco/converters/tensor.py:54
↓ 5 callersMethodscaled_dot_product_attention
(self, q, k, v)
examples/multihead-attention.py:80
↓ 5 callersMethodstyles_join
(self, style1: Dict, style2: Dict)
nobuco/vis/html_stylizer.py:72
↓ 5 callersMethodvalidation_status_to_style
(self, status: ValidationStatus, converted_manually)
nobuco/vis/console_stylizer.py:16
↓ 4 callersFunction_flatten
(iterable)
nobuco/converters/tensor.py:85
↓ 4 callersMethodadd
(self, tensor: torch.Tensor)
nobuco/trace/tensor_storage.py:18
↓ 4 callersMethodbuild
(node_list: Collection[PytorchNode])
nobuco/trace/trace.py:438
↓ 4 callersFunctionclone_torch_tensors_recursively_with_cache
(obj, storage: TensorStorage)
nobuco/trace/tensor_storage.py:25
↓ 4 callersFunctioncreate_layer
(i, reverse)
nobuco/node_converters/recurrent.py:45
↓ 4 callersMethodop_tracing_decorator
(orig_method, op_cls, module_suffix=None, is_whitelist_op=False, need_trace_deeper=True)
nobuco/trace/trace.py:286
↓ 4 callersFunctionperm_compose
(perm_next, perm_prev)
nobuco/converters/tensor.py:40
↓ 4 callersFunctionreorder
(param)
nobuco/node_converters/recurrent.py:47
↓ 4 callersFunctionset_torch_tensor_id
(tensor, id)
nobuco/util.py:28
↓ 4 callersFunctiontf_annotate_recursively
(obj, channel_order)
nobuco/layers/channel_order.py:101
↓ 3 callersMethod__init__
Full Mamba model.
examples/mamba_minimal/model.py:61
↓ 3 callersMethod__str__
(self)
nobuco/layers/stub.py:9
↓ 3 callersFunctionbroadcast
(tensors)
nobuco/node_converters/tensor_broadcast.py:12
↓ 3 callersFunctioncollect_recursively_func
(obj, predicate: Callable[[object], bool])
nobuco/util.py:32
↓ 3 callersFunctionget_link_to_obj
Print a link in PyCharm to a module, function, class, method or property.
nobuco/locate/link.py:17
↓ 3 callersMethodget_name
(self, tensor)
nobuco/entity/pytorch.py:42
↓ 3 callersFunctionpt2tf
(t, dtype=None)
examples/stablelm_zephyr/run_tf.py:18
↓ 3 callersMethodsplit_heads
(self, x, batch_size)
examples/multihead-attention.py:55
↓ 3 callersFunctionsummarize_tensors
(obj)
nobuco/converters/validation.py:30
↓ 3 callersFunctiont_pytorch2keras
(tensor_pt, channel_order=ChannelOrder.PYTORCH)
nobuco/converters/channel_ordering.py:27
↓ 3 callersFunctionunwrap_torch_tensors_recursively
(obj, annotate=True)
nobuco/trace/trace.py:74
↓ 2 callersMethod__init__
(self, shape, alpha)
examples/stateful3.py:18
↓ 2 callersMethod_traverse_graph
(self, start_names, op_descr_list, reverse_graph)
nobuco/layers/container.py:26
↓ 2 callersMethodall_tensors_equal
(tensors1, tensors2)
nobuco/trace/trace.py:235
↓ 2 callersMethodappend_node
(node: PytorchNode)
nobuco/trace/trace.py:220
↓ 2 callersMethodbranch_false
(self, x)
examples/control_if2.py:32
↓ 2 callersMethodbranch_true
(self, x)
examples/control_if2.py:25
↓ 2 callersFunctionchannel_interleave2d
(x, block_size: int, reverse: bool)
nobuco/node_converters/pooling.py:80
↓ 2 callersFunctionchannel_interleave2d
(x: torch.Tensor, block_size: int, reverse: bool)
nobuco/addons/torch/util.py:4
↓ 2 callersMethodclose
(self)
nobuco/vis/progress.py:29
↓ 2 callersMethodcond
(self, inputs)
examples/control_if2.py:39
↓ 2 callersFunctionconvert_initial_states
(hx)
nobuco/node_converters/recurrent.py:82
↓ 2 callersFunctioncount_nonnone
(slices)
nobuco/node_converters/slice.py:23
↓ 2 callersMethodcreate
(cls, weight, trainable)
nobuco/layers/weight.py:24
↓ 2 callersMethoddecorate_all
()
nobuco/trace/trace.py:422
↓ 2 callersFunctiondims_pytorch2keras
(dims, num_dims)
nobuco/converters/tensor.py:58
↓ 2 callersFunctiondivide
(x, y, rounding_mode)
nobuco/node_converters/math.py:75
↓ 2 callersFunctionfind_call_summary
(call_stack: List[traceback.FrameSummary])
nobuco/trace/trace.py:99
↓ 2 callersMethodforward_stateless
(self, x, x_ema)
examples/stateful2.py:25
↓ 2 callersFunctionfunc
(x)
nobuco/converters/tensor.py:80
↓ 2 callersMethodget_op_count
(self)
nobuco/entity/pytorch.py:120
↓ 2 callersFunctionhard_sigmoid_pytorch_compatible
(x)
nobuco/node_converters/activation.py:18
↓ 2 callersFunctionhas_converter
(node: PytorchNode, converter_dict: Dict[object, Pytorch2KerasNodeConverter])
nobuco/convert.py:44
↓ 2 callersFunctionindices_make_non_negative
(indices, d)
nobuco/node_converters/slice.py:241
↓ 2 callersFunctioninner_func
(input, dim, keepdim)
nobuco/node_converters/reduce.py:48
↓ 2 callersMethodis_connected
(self)
nobuco/commons.py:49
↓ 2 callersFunctionis_identity_perm
(perm)
nobuco/converters/tensor.py:70
↓ 2 callersMethodis_tracing_enabled
()
nobuco/trace/trace.py:216
↓ 2 callersFunctionmake_template_recursively
(obj)
nobuco/converters/channel_ordering.py:62
↓ 2 callersMethodmodule_forward_tracing_decorator
(forward_func)
nobuco/trace/trace.py:244
↓ 2 callersFunctionperm_identity
(n_dims)
nobuco/converters/tensor.py:29
↓ 2 callersFunctionprelu_prepare_params
(weight_np, input_dim, input_channels, input_channel_order)
nobuco/node_converters/activation.py:85
↓ 2 callersFunctionprepare_inputs_pt
(prompt, tokenizer, past_kv_flattened, device='cpu')
examples/stablelm_zephyr/util.py:4
↓ 2 callersFunctionprocess_coord
(grid, w_h)
nobuco/node_converters/grid_sampling.py:43
↓ 2 callersFunctionpytorch2keras_recursively
(obj, channel_order=ChannelOrder.PYTORCH)
nobuco/converters/channel_ordering.py:35
↓ 2 callersFunctionreplace_func
(obj)
nobuco/util.py:76
↓ 2 callersFunctionreplace_func
(obj)
nobuco/trace/trace.py:61
↓ 2 callersFunctionslices_adjust
(slices_enumerated, start_i, delta)
nobuco/node_converters/slice.py:273
↓ 2 callersFunctionslices_make_full
(slices, n_dims)
nobuco/node_converters/slice.py:20
↓ 2 callersMethodstylize
(self, text: str, style: Dict)
nobuco/vis/html_stylizer.py:21
↓ 2 callersFunctiont_keras2pytorch
(tensor_tf, restore_channel_order=False)
nobuco/converters/channel_ordering.py:20
↓ 2 callersFunctiontemplate_insert_recursively
(obj, tensors)
nobuco/converters/channel_ordering.py:77
↓ 2 callersFunctiontf_assert_has_attr_recursively
(obj, attr)
nobuco/layers/channel_order.py:113
↓ 2 callersFunctiontf_set_order_recursively
(obj, channel_order: ChannelOrder)
nobuco/layers/channel_order.py:83
↓ 2 callersFunctionto_shape_and_dtype
(assigned_tensor, shape, dtype)
nobuco/node_converters/slice.py:44
↓ 1 callersMethod__init__
(self, body, *args, **kwargs)
examples/control_for.py:44
↓ 1 callersMethod__init__
(self, d_model, num_heads)
examples/multihead-attention.py:28
↓ 1 callersMethod__init__
(self)
examples/stateful1.py:18
↓ 1 callersMethod__init__
(self)
examples/control_if.py:17
next →1–100 of 654, ranked by callers