Path	Lines of Code
MANIFEST.in	2
build_deps/tf_dependency/BUILD.tpl	14
build_deps/tf_dependency/build_defs.bzl.tpl	2
build_deps/tf_dependency/tf_configure.bzl	205
build_deps/toolchains/gcc7_manylinux2010-nvcc-cuda11/cc_toolchain_config.bzl	1411
build_deps/toolchains/gpu/crosstool/BUILD.tpl	60
build_deps/toolchains/gpu/crosstool/CROSSTOOL.tpl	1304
build_deps/toolchains/gpu/crosstool/cc_toolchain_config.bzl.tpl	1411
build_deps/toolchains/gpu/crosstool/windows/msvc_wrapper_for_nvcc.py.tpl	153
build_deps/toolchains/gpu/cuda/BUILD.tpl	204
build_deps/toolchains/gpu/cuda/BUILD.windows.tpl	143
build_deps/toolchains/gpu/cuda/build_defs.bzl.tpl	50
build_deps/toolchains/gpu/cuda/cuda_config.h.tpl	18
build_deps/toolchains/gpu/cuda_configure.bzl	1014
build_deps/toolchains/gpu/find_cuda_config.py	455
configure.py	106
pyproject.toml	18
setup.py	83
tensorflow_addons/__init__.py	16
tensorflow_addons/activations/__init__.py	9
tensorflow_addons/activations/gelu.py	13
tensorflow_addons/activations/hardshrink.py	17
tensorflow_addons/activations/lisht.py	7
tensorflow_addons/activations/mish.py	7
tensorflow_addons/activations/rrelu.py	28
tensorflow_addons/activations/snake.py	8
tensorflow_addons/activations/softshrink.py	15
tensorflow_addons/activations/sparsemax.py	59
tensorflow_addons/activations/tanhshrink.py	7
tensorflow_addons/callbacks/__init__.py	3
tensorflow_addons/callbacks/average_model_checkpoint.py	54
tensorflow_addons/callbacks/time_stopping.py	33
tensorflow_addons/callbacks/tqdm_progress_bar.py	170
tensorflow_addons/conftest.py	24
tensorflow_addons/custom_ops/image/cc/kernels/adjust_hsv_in_yiq_op.cc	138
tensorflow_addons/custom_ops/image/cc/kernels/adjust_hsv_in_yiq_op.h	53
tensorflow_addons/custom_ops/image/cc/kernels/adjust_hsv_in_yiq_op_gpu.cu.cc	59
tensorflow_addons/custom_ops/image/cc/kernels/connected_components.cc	92
tensorflow_addons/custom_ops/image/cc/kernels/connected_components.h	200
tensorflow_addons/custom_ops/image/cc/kernels/euclidean_distance_transform_op.cc	92
tensorflow_addons/custom_ops/image/cc/kernels/euclidean_distance_transform_op.h	109
tensorflow_addons/custom_ops/image/cc/kernels/euclidean_distance_transform_op_gpu.cu.cc	47
tensorflow_addons/custom_ops/image/cc/kernels/resampler_ops.cc	318
tensorflow_addons/custom_ops/image/cc/kernels/resampler_ops.h	30
tensorflow_addons/custom_ops/image/cc/kernels/resampler_ops_gpu.cu.cc	191
tensorflow_addons/custom_ops/image/cc/ops/distort_image_ops.cc	48
tensorflow_addons/custom_ops/image/cc/ops/image_ops.cc	48
tensorflow_addons/custom_ops/image/cc/ops/resampler_ops.cc	41
tensorflow_addons/custom_ops/layers/cc/kernels/correlation_cost_op.cc	295
tensorflow_addons/custom_ops/layers/cc/kernels/correlation_cost_op.h	26
tensorflow_addons/custom_ops/layers/cc/kernels/correlation_cost_op_gpu.cu.cc	353
tensorflow_addons/custom_ops/layers/cc/kernels/embedding_bag_backward_kernels.cu.cc	184
tensorflow_addons/custom_ops/layers/cc/kernels/embedding_bag_ops.cc	273
tensorflow_addons/custom_ops/layers/cc/kernels/embedding_bag_ops.h	34
tensorflow_addons/custom_ops/layers/cc/kernels/embedding_bag_ops_gpu.cu.cc	66
tensorflow_addons/custom_ops/layers/cc/ops/correlation_cost_op.cc	98
tensorflow_addons/custom_ops/layers/cc/ops/embedding_bag_ops.cc	50
tensorflow_addons/custom_ops/seq2seq/cc/kernels/beam_search_ops.cc	157
tensorflow_addons/custom_ops/seq2seq/cc/kernels/beam_search_ops.h	21
tensorflow_addons/custom_ops/seq2seq/cc/kernels/beam_search_ops_gpu.cu.cc	80
tensorflow_addons/custom_ops/seq2seq/cc/ops/beam_search_ops.cc	46
tensorflow_addons/custom_ops/text/cc/kernels/parse_time_kernel.cc	76
tensorflow_addons/custom_ops/text/cc/kernels/skip_gram_kernels.cc	105
tensorflow_addons/custom_ops/text/cc/ops/parse_time_op.cc	45
tensorflow_addons/custom_ops/text/cc/ops/skip_gram_ops.cc	34
tensorflow_addons/image/__init__.py	26
tensorflow_addons/image/color_ops.py	65
tensorflow_addons/image/compose_ops.py	18
tensorflow_addons/image/connected_components.py	40
tensorflow_addons/image/cutout_ops.py	104
tensorflow_addons/image/dense_image_warp.py	133
tensorflow_addons/image/distance_transform.py	22
tensorflow_addons/image/distort_image_ops.py	113
tensorflow_addons/image/filters.py	142
tensorflow_addons/image/interpolate_spline.py	113
tensorflow_addons/image/resampler_ops.py	19
tensorflow_addons/image/sparse_image_warp.py	114
tensorflow_addons/image/transform_ops.py	168
tensorflow_addons/image/translate_ops.py	58
tensorflow_addons/image/utils.py	73
tensorflow_addons/layers/__init__.py	30
tensorflow_addons/layers/adaptive_pooling.py	209
tensorflow_addons/layers/crf.py	145
tensorflow_addons/layers/embedding_bag.py	87
tensorflow_addons/layers/esn.py	110
tensorflow_addons/layers/gelu.py	18
tensorflow_addons/layers/max_unpooling_2d.py	89
tensorflow_addons/layers/max_unpooling_2d_v2.py	40
tensorflow_addons/layers/maxout.py	42
tensorflow_addons/layers/multihead_attention.py	169
tensorflow_addons/layers/netvlad.py	56
tensorflow_addons/layers/noisy_dense.py	187
tensorflow_addons/layers/normalizations.py	334
tensorflow_addons/layers/optical_flow.py	142
tensorflow_addons/layers/poincare.py	25
tensorflow_addons/layers/polynomial.py	117
tensorflow_addons/layers/snake.py	23
tensorflow_addons/layers/sparsemax.py	18
tensorflow_addons/layers/spatial_pyramid_pooling.py	66
tensorflow_addons/layers/spectral_normalization.py	62
tensorflow_addons/layers/stochastic_depth.py	28
tensorflow_addons/layers/tlu.py	76
tensorflow_addons/layers/wrappers.py	127
tensorflow_addons/losses/__init__.py	22
tensorflow_addons/losses/contrastive.py	28
tensorflow_addons/losses/focal_loss.py	52
tensorflow_addons/losses/giou_loss.py	55
tensorflow_addons/losses/kappa_loss.py	59
tensorflow_addons/losses/lifted.py	77
tensorflow_addons/losses/metric_learning.py	38
tensorflow_addons/losses/npairs.py	42
tensorflow_addons/losses/quantiles.py	26
tensorflow_addons/losses/sparsemax_loss.py	57
tensorflow_addons/losses/triplet.py	187
tensorflow_addons/metrics/__init__.py	18
tensorflow_addons/metrics/cohens_kappa.py	125
tensorflow_addons/metrics/f_scores.py	121
tensorflow_addons/metrics/geometric_mean.py	38
tensorflow_addons/metrics/hamming.py	44
tensorflow_addons/metrics/harmonic_mean.py	15
tensorflow_addons/metrics/kendalls_tau.py	128
tensorflow_addons/metrics/matthews_correlation_coefficient.py	60
tensorflow_addons/metrics/multilabel_confusion_matrix.py	87
tensorflow_addons/metrics/r_square.py	119
tensorflow_addons/metrics/utils.py	42
tensorflow_addons/optimizers/__init__.py	34
tensorflow_addons/optimizers/adabelief.py	203
tensorflow_addons/optimizers/average_wrapper.py	112
tensorflow_addons/optimizers/cocob.py	63
tensorflow_addons/optimizers/conditional_gradient.py	124
tensorflow_addons/optimizers/cyclical_learning_rate.py	127
tensorflow_addons/optimizers/discriminative_layer_training.py	77
tensorflow_addons/optimizers/lamb.py	153
tensorflow_addons/optimizers/lazy_adam.py	62
tensorflow_addons/optimizers/lookahead.py	107
tensorflow_addons/optimizers/moving_average.py	109
tensorflow_addons/optimizers/novograd.py	156
tensorflow_addons/optimizers/proximal_adagrad.py	89
tensorflow_addons/optimizers/rectified_adam.py	209
tensorflow_addons/optimizers/stochastic_weight_averaging.py	44
tensorflow_addons/optimizers/utils.py	38
tensorflow_addons/optimizers/weight_decay_optimizers.py	155
tensorflow_addons/optimizers/yogi.py	193
tensorflow_addons/options.py	24
tensorflow_addons/register.py	38
tensorflow_addons/rnn/__init__.py	5
tensorflow_addons/rnn/esn_cell.py	125
tensorflow_addons/rnn/layer_norm_lstm_cell.py	109
tensorflow_addons/rnn/layer_norm_simple_rnn_cell.py	98
tensorflow_addons/rnn/nas_cell.py	121
tensorflow_addons/rnn/peephole_lstm_cell.py	54
tensorflow_addons/seq2seq/__init__.py	32
tensorflow_addons/seq2seq/attention_wrapper.py	1035
tensorflow_addons/seq2seq/basic_decoder.py	68
tensorflow_addons/seq2seq/beam_search_decoder.py	737
tensorflow_addons/seq2seq/decoder.py	294
tensorflow_addons/seq2seq/loss.py	122
tensorflow_addons/seq2seq/sampler.py	498
tensorflow_addons/tensorflow_addons.bzl	64
tensorflow_addons/testing/__init__.py	1
tensorflow_addons/testing/serialization.py	77
tensorflow_addons/text/__init__.py	19
tensorflow_addons/text/crf.py	291
tensorflow_addons/text/crf_wrapper.py	93
tensorflow_addons/text/parse_time_op.py	10
tensorflow_addons/text/skip_gram_ops.py	194
tensorflow_addons/utils/__init__.py	1
tensorflow_addons/utils/ensure_tf_install.py	35
tensorflow_addons/utils/keras_utils.py	100
tensorflow_addons/utils/resource_loader.py	80
tensorflow_addons/utils/types.py	39
tensorflow_addons/version.py	9
tools/docker/build_wheel.Dockerfile	44
tools/docker/dev_container.Dockerfile	23
tools/docker/pre-commit.Dockerfile	9
tools/docker/sanity_check.Dockerfile	112
tools/format.py	36