MCPcopy Create free account

hub / github.com/AlexanderLutsenko/nobuco / functions

Functions654 in github.com/AlexanderLutsenko/nobuco

⨍ Functions 654◇ Types & classes 95↳ Endpoints 3
↓ 1 callersMethod__init__
(self)
examples/pixelshuffle.py:21
↓ 1 callersMethod__init__
(self, cond, body, *args, **kwargs)
examples/control_while.py:48
↓ 1 callersMethod__init__
(self)
examples/stateful2.py:18
↓ 1 callersMethod__init__
(self, branch_true, branch_false, *args, **kwargs)
examples/control_if2.py:54
↓ 1 callersFunction_dims_make_positive
(dims, n_dims, add_one=False)
nobuco/converters/tensor.py:12
↓ 1 callersFunction_ensure_tuple
(iterable)
nobuco/converters/tensor.py:101
↓ 1 callersMethodapply_module_tracing_recursively
(module)
nobuco/trace/trace.py:427
↓ 1 callersMethodbn_conv
(self, x)
examples/fusible.py:39
↓ 1 callersMethodbuild_hierarchy
(node_list: Collection[PytorchNode])
nobuco/trace/trace.py:436
↓ 1 callersFunctioncalc_diff
(t1, t2)
nobuco/converters/validation.py:129
↓ 1 callersFunctioncalc_diff_boolean
(t1, t2)
nobuco/converters/validation.py:166
↓ 1 callersFunctioncalc_diff_numerical
(t1, t2)
nobuco/converters/validation.py:130
↓ 1 callersFunctionclone
(tensor)
nobuco/trace/tensor_storage.py:28
↓ 1 callersFunctionclone_torch_tensors_recursively
(obj, annotate=True)
nobuco/util.py:73
↓ 1 callersFunctioncollect
(obj)
nobuco/util.py:36
↓ 1 callersFunctioncollect_conversion_results
(keras_node: KerasConvertedNode)
nobuco/convert.py:218
↓ 1 callersFunctioncollect_disconnected_node_names
(input_names, output_names, keras_converted_nodes)
nobuco/convert.py:91
↓ 1 callersFunctioncollect_take_indices_and_specs
(slices_enumerated)
nobuco/node_converters/slice.py:230
↓ 1 callersFunctioncollect_tensors_by_ids
(tensor_ids: Collection[int], node_hierarchies: Collection[PytorchNodeHierarchy], output_tensors)
nobuco/convert.py:105
↓ 1 callersFunctioncollect_validation_results
(keras_node: KerasConvertedNode)
nobuco/convert.py:211
↓ 1 callersFunctioncond
(self, inputs)
examples/control_if2.py:77
↓ 1 callersMethodconnectivity_status_to_style
(self, tensor, self_status: ConnectivityStatus, parent_status: ConnectivityStatus, is_input)
nobuco/vis/html_stylizer.py:53
↓ 1 callersFunctionconvert
(hierarchy: PytorchNodeHierarchy, converted_op_dict: Dict, reuse_layers: bool, full_validation: bool, depth)
nobuco/convert.py:138
↓ 1 callersFunctionconvert_container
( node: PytorchNode, children: Collection[PytorchNodeHierarchy], children_converted_no
nobuco/convert.py:83
↓ 1 callersFunctionconvert_hierarchy
( node_hierarchy: PytorchNodeHierarchy, converter_dict: Dict[object, Pytorch2KerasNodeConverte
nobuco/convert.py:128
↓ 1 callersFunctionconvert_node
(node: PytorchNode, node_converter: Pytorch2KerasNodeConverter)
nobuco/convert.py:70
↓ 1 callersMethoddecorate_module
()
nobuco/trace/trace.py:375
↓ 1 callersMethoddecorate_ops
()
nobuco/trace/trace.py:403
↓ 1 callersFunctionenumerate_properly
(slices, n_dims)
nobuco/node_converters/slice.py:214
↓ 1 callersFunctionfind_converter
(node: PytorchNode, converter_dict: Dict[object, Pytorch2KerasNodeConverter])
nobuco/convert.py:48
↓ 1 callersFunctionfind_unimplemented
(hierarchy: PytorchNodeHierarchy, converter_dict: Dict[object, Pytorch2KerasNodeConverter])
nobuco/convert.py:52
↓ 1 callersMethodfor_loop
(self, x, n)
examples/control_for.py:31
↓ 1 callersMethodforward_same
(self, x)
examples/module_deepcopy.py:38
↓ 1 callersFunctionget_broadcast_shape
(sizes, tensor_shape)
nobuco/node_converters/tensor_manipulation.py:179
↓ 1 callersMethodget_connectivity_status
(self)
nobuco/layers/container.py:47
↓ 1 callersMethodget_converter_link
(self)
nobuco/converters/validation.py:50
↓ 1 callersFunctionget_link
Print a link in PyCharm to a line in file. Defaults to line where this function was called.
nobuco/locate/link.py:6
↓ 1 callersFunctionget_type
(op)
nobuco/util.py:90
↓ 1 callersFunctiongetitem_sophisticated
(self, n_dims, *slices)
nobuco/node_converters/slice.py:209
↓ 1 callersFunctionhandle_ellipsis
(slices, n_dims)
nobuco/node_converters/slice.py:26
↓ 1 callersFunctionhard_swish_pytorch_compatible
(x)
nobuco/node_converters/activation.py:23
↓ 1 callersFunctionhard_tanh_pytorch_compatible
(x, min_val, max_val)
nobuco/node_converters/activation.py:28
↓ 1 callersFunctionis_light
(slices)
nobuco/node_converters/slice.py:320
↓ 1 callersMethodis_module
(self)
nobuco/entity/pytorch.py:111
↓ 1 callersFunctionl2_dist
(A, B)
nobuco/node_converters/math.py:306
↓ 1 callersFunctionloop
(self, x)
examples/control_while.py:69
↓ 1 callersMethodloop
(self, x)
examples/control_while.py:35
↓ 1 callersMethodloop_body
(self, x)
examples/control_for.py:24
↓ 1 callersMethodloop_body
(self, x)
examples/control_while.py:28
↓ 1 callersMethodloop_cond
(self, x)
examples/control_while.py:24
↓ 1 callersMethodmake_inputs_template
(self)
nobuco/entity/pytorch.py:72
↓ 1 callersMethodmake_outputs_template
(self)
nobuco/entity/pytorch.py:76
↓ 1 callersFunctionperm_invert
(perm)
nobuco/converters/tensor.py:33
↓ 1 callersFunctionpermute_keras2pytorch
(xs)
nobuco/converters/tensor.py:66
↓ 1 callersMethodpostprocess
(self, text)
nobuco/vis/console_stylizer.py:10
↓ 1 callersFunctionpostprocess_outputs_tf
(outputs, outputs_channel_order, output_names)
nobuco/convert.py:251
↓ 1 callersFunctionpredicate
(obj)
nobuco/util.py:61
↓ 1 callersFunctionprepare_inputs_tf
(inputs_pt, inputs_channel_order, input_shapes, input_names)
nobuco/convert.py:225
↓ 1 callersFunctionprint_trainable_parameters
(model)
examples/module_deepcopy.py:68
↓ 1 callersFunctionreplace_func
(tensor)
nobuco/trace/tensor_storage.py:34
↓ 1 callersMethodreset_states
(self)
nobuco/layers/container.py:80
↓ 1 callersMethodreset_states
(self)
examples/stateful1.py:60
↓ 1 callersMethodreset_states
(self)
examples/stateful2.py:57
↓ 1 callersMethodreset_states
(self)
examples/stateful3.py:66
↓ 1 callersFunctionrun_tflite
(interpreter, *np_inputs)
examples/stateful1.py:117
↓ 1 callersFunctionrun_tflite
(interpreter, *np_inputs)
examples/stateful2.py:114
↓ 1 callersFunctionrun_tflite
(interpreter, *np_inputs)
examples/stateful3.py:119
↓ 1 callersMethodselective_scan
Does selective scan algorithm. See: - Section 2 State Space Models in the Mamba paper [1] - Algorithm 2 in Section 3.2 in the
examples/mamba_minimal/model.py:287
↓ 1 callersMethodselective_scan_loop
(self, l, x, deltaA, deltaB_u, C)
examples/mamba_minimal/model.py:278
↓ 1 callersFunctionslice_assign
(sliced_tensor, slice_args, assigned_tensor)
nobuco/node_converters/slice.py:51
↓ 1 callersMethodssm
Runs the SSM. See: - Algorithm 2 in Section 3.2 in the Mamba paper [1] - run_SSM(A, B, C, u) in The Annotated S4 [2]
examples/mamba_minimal/model.py:242
↓ 1 callersFunctionstd
(input, axis, unbiased=True, keepdims=False)
nobuco/node_converters/reduce.py:85
↓ 1 callersMethodstyles_join
(self, style1: str, style2: str)
nobuco/vis/console_stylizer.py:49
↓ 1 callersFunctiontake_nd
(x, indices_list, start_dim)
nobuco/node_converters/slice.py:246
↓ 1 callersFunctiontake_nd_disjoint
(x, dims, indices_list, n_dims)
nobuco/node_converters/slice.py:261
↓ 1 callersFunctiontf_autocast_recursively
(inputs, type_priority_list=None)
nobuco/converters/type_cast.py:57
↓ 1 callersFunctiontf_cast_recursively
(obj, types)
nobuco/converters/type_cast.py:88
↓ 1 callersFunctiontf_pad_circular
(x, pad)
nobuco/node_converters/padding.py:30
↓ 1 callersFunctiontf_pad_replicate
(x, pad)
nobuco/node_converters/padding.py:15
↓ 1 callersFunctiontf_scalars_to_tensors_recursively
(inputs)
nobuco/converters/scalars_to_tensors.py:10
↓ 1 callersFunctiontorch_gather
See: https://stackoverflow.com/a/71037111/4850610
nobuco/node_converters/tensor_manipulation.py:304
↓ 1 callersMethodtrace
(module_or_function: nn.Module | Callable, trace_shape: bool, enable_torch_tracing: bool, args, kwargs, progre
nobuco/trace/trace.py:455
↓ 1 callersFunctiontraceable
(func_to_trace: Callable)
nobuco/trace/trace.py:90
↓ 1 callersFunctiontril
(h, w)
nobuco/node_converters/attention.py:72
↓ 1 callersFunctionvalidate
(node, pytorch_op, keras_op, input_args, input_kwargs, output_tensors, op_type, tolerance=1e-4)
nobuco/converters/validation.py:66
↓ 1 callersFunctionvalidate_diff_default
(keras_op, pytorch_op, args_pt, kwargs_pt, outputs_pt, is_training=False)
nobuco/converters/validation.py:107
↓ 1 callersFunctionvar_unbiased
(input, axis, keepdims=False)
nobuco/node_converters/reduce.py:72
Method__call__
(self, x, initial_state=None)
nobuco/node_converters/recurrent.py:18
Method__call__
(self, *args, **kwargs)
nobuco/layers/stub.py:6
Method__call__
(self, *args, **kwargs)
nobuco/layers/stub.py:17
Method__call__
(self, *args, **kwargs)
nobuco/layers/channel_order.py:16
Method__call__
(self, *args, training=False, **kwargs)
nobuco/layers/container.py:58
Method__del__
(self)
nobuco/vis/progress.py:36
Method__enter__
(self)
nobuco/vis/progress.py:17
Method__enter__
(self)
nobuco/trace/trace.py:165
Method__enter__
(self)
nobuco/trace/trace.py:179
Method__enter__
(self)
nobuco/trace/trace.py:194
Method__enter__
(self)
nobuco/trace/trace.py:209
Method__exit__
(self, exc_type, exc_val, exc_tb)
nobuco/vis/progress.py:20
Method__exit__
(self, *args)
nobuco/trace/trace.py:170
← previousnext →101–200 of 654, ranked by callers