path	# lines of code
codecov.yml	14
setup.py	147
benchmarks/classy.py	67
benchmarks/matmul.py	26
benchmarks/binary.py	37
benchmarks/mha_cuda.py	56
benchmarks/unbind.py	34
benchmarks/embedding.py	48
benchmarks/linear.py	45
benchmarks/jit_tensorwise.py	30
benchmarks/nearest_neighbors.py	114
benchmarks/jit_apply.py	51
benchmarks/conv2d.py	45
benchmarks/basic.py	16
benchmarks/utils.py	48
benchmarks/unary.py	28
benchmarks/segmentation_layers.py	291
benchmarks/gat.py	90
benchmarks/mha.py	91
benchmarks/segmentation.py	27
nestedtensor/csrc/mha.cpp	68
nestedtensor/csrc/ReduceOps.cpp	324
nestedtensor/csrc/autograd_functions.cpp	202
nestedtensor/csrc/BinaryOps.cpp	532
nestedtensor/csrc/python_functions.cpp	257
nestedtensor/csrc/py_init.cpp	229
nestedtensor/csrc/nested_tensor_impl.cpp	395
nestedtensor/csrc/matmul.cpp	51
nestedtensor/csrc/conv2d.cpp	196
nestedtensor/csrc/py_utils.cpp	7
nestedtensor/csrc/cuda/mha.cpp	89
nestedtensor/csrc/cuda/attention.h	33
nestedtensor/csrc/cuda/layernorm.cpp	64
nestedtensor/csrc/cuda/layernorm.h	16
nestedtensor/csrc/cuda/attention.cu	305
nestedtensor/csrc/cuda/transformer_kernels.h	23
nestedtensor/csrc/cuda/add.h	57
nestedtensor/csrc/cuda/add.cu	290
nestedtensor/csrc/cuda/transformer_kernels.cu	305
nestedtensor/csrc/cuda/common.h	50
nestedtensor/csrc/cuda/padding.cu	340
nestedtensor/csrc/cuda/cuda_kernels.h	46
nestedtensor/csrc/cuda/transpose.cu	240
nestedtensor/csrc/cuda/transpose.h	29
nestedtensor/csrc/cuda/cuda_kernels.cu	437
nestedtensor/csrc/cuda/padding.h	41
nestedtensor/csrc/totensor.cpp	66
nestedtensor/csrc/transpose.cpp	188
nestedtensor/csrc/python_functions.h	6
nestedtensor/csrc/UnaryOps.cpp	179
nestedtensor/csrc/pooling.cpp	77
nestedtensor/csrc/utils/nested_node.h	372
nestedtensor/csrc/utils/nested_node.cpp	1
nestedtensor/csrc/utils/python_nested_node.h	94
nestedtensor/csrc/utils/nested_node_functions.h	89
nestedtensor/csrc/utils/python_nested_node.cpp	109
nestedtensor/csrc/creation.h	15
nestedtensor/csrc/storage/Packed.h	269
nestedtensor/csrc/storage/StorageBase.h	46
nestedtensor/csrc/storage/common.h	34
nestedtensor/csrc/storage/EfficientSizeNode.h	212
nestedtensor/csrc/storage/Storage.h	2
nestedtensor/csrc/masking.cpp	547
nestedtensor/csrc/python_args.h	43
nestedtensor/csrc/EmbeddingBag.cpp	47
nestedtensor/csrc/shape.cpp	82
nestedtensor/csrc/py_utils.h	51
nestedtensor/csrc/creation.cpp	193
nestedtensor/csrc/SoftMax.cpp	47
nestedtensor/csrc/scripts/binaryops.py	113
nestedtensor/csrc/nested_tensor_impl.h	284
nestedtensor/csrc/fold.cpp	46
nestedtensor/csrc/activation.cpp	62
nestedtensor/csrc/masking.h	30
nestedtensor/csrc/functions.cpp	252
nestedtensor/csrc/ComparisonOps.cpp	25
nestedtensor/csrc/BinaryOps.h	65
nestedtensor/csrc/transpose.h	11
nestedtensor/csrc/Expand.cpp	40
nestedtensor/version.py	5
nestedtensor/nn/__init__.py	1
nestedtensor/nn/mha.py	71
nestedtensor/__init__.py	15
nestedtensor/nested/fuser.py	203
nestedtensor/nested/creation.py	24
nestedtensor/nested/__init__.py	1
nestedtensor/nested/masking.py	33
nestedtensor/nested/nested.py	330