tensorflow/models

View on GitHub
official/projects/edgetpu/vision/modeling/optimized_multiheadattention_layer_test.py

Summary

Maintainability
A
1 hr
Test Coverage

Function test_same_output has 27 lines of code (exceeds 25 allowed). Consider refactoring.
Open

  def test_same_output(self):
    """Tests that OptimizedMultiHeadAttention returns the expected outputs."""

    input_tensor_1 = tf.random.uniform((_BATCH_SIZE, _SEQ_LEN, _EMBEDDING_SIZE))
    input_tensor_2 = tf.random.uniform((_BATCH_SIZE, _SEQ_LEN, _EMBEDDING_SIZE))

    There are no issues that match your filters.

    Category
    Status