Commit 658c8038 authored by Frederick Liu's avatar Frederick Liu Committed by A. Unique TensorFlower
Browse files

[bigbird] move bigbird from nlp/projects to official/projects

PiperOrigin-RevId: 418851251
parent a7e60974
...@@ -16,9 +16,9 @@ ...@@ -16,9 +16,9 @@
Includes configurations and factory methods. Includes configurations and factory methods.
""" """
import dataclasses
from typing import Optional from typing import Optional
import dataclasses
import gin import gin
import tensorflow as tf import tensorflow as tf
...@@ -26,7 +26,7 @@ from official.modeling import hyperparams ...@@ -26,7 +26,7 @@ from official.modeling import hyperparams
from official.modeling import tf_utils from official.modeling import tf_utils
from official.nlp.modeling import layers from official.nlp.modeling import layers
from official.nlp.modeling import networks from official.nlp.modeling import networks
from official.nlp.projects.bigbird import encoder as bigbird_encoder from official.projects.bigbird import encoder as bigbird_encoder
@dataclasses.dataclass @dataclasses.dataclass
......
...@@ -20,8 +20,8 @@ import tensorflow as tf ...@@ -20,8 +20,8 @@ import tensorflow as tf
from official.modeling import activations from official.modeling import activations
from official.nlp import modeling from official.nlp import modeling
from official.nlp.modeling import layers from official.nlp.modeling import layers
from official.nlp.projects.bigbird import recompute_grad from official.projects.bigbird import recompute_grad
from official.nlp.projects.bigbird import recomputing_dropout from official.projects.bigbird import recomputing_dropout
_MAX_SEQ_LEN = 4096 _MAX_SEQ_LEN = 4096
......
...@@ -17,7 +17,7 @@ ...@@ -17,7 +17,7 @@
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from official.nlp.projects.bigbird import encoder from official.projects.bigbird import encoder
class BigBirdEncoderTest(tf.test.TestCase): class BigBirdEncoderTest(tf.test.TestCase):
......
...@@ -17,8 +17,8 @@ ...@@ -17,8 +17,8 @@
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
from official.nlp.projects.bigbird import recompute_grad as recompute_grad_lib from official.projects.bigbird import recompute_grad as recompute_grad_lib
from official.nlp.projects.bigbird import stateless_dropout as stateless_dropout_lib from official.projects.bigbird import stateless_dropout as stateless_dropout_lib
# Reimplements internal function # Reimplements internal function
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment