File tree Expand file tree Collapse file tree 18 files changed +0
-28
lines changed
tensorflow_probability/python/experimental/auto_batching Expand file tree Collapse file tree 18 files changed +0
-28
lines changed Original file line number Diff line number Diff line change 14
14
# ============================================================================
15
15
"""Tests of the allocation strategy optimization pass."""
16
16
17
- # Dependency imports
18
-
19
17
from tensorflow_probability .python .experimental .auto_batching import allocation_strategy
20
18
from tensorflow_probability .python .experimental .auto_batching import instructions as inst
21
19
from tensorflow_probability .python .experimental .auto_batching import test_programs
Original file line number Diff line number Diff line change 24
24
import contextlib
25
25
import inspect
26
26
27
- # Dependency imports
28
-
29
27
from tensorflow_probability .python .experimental .auto_batching import instructions as inst
30
28
31
29
__all__ = [
Original file line number Diff line number Diff line change 12
12
# See the License for the specific language governing permissions and
13
13
# limitations under the License.
14
14
# ============================================================================
15
- # Dependency imports
16
15
import numpy as np
17
16
18
17
import tensorflow .compat .v2 as tf
Original file line number Diff line number Diff line change 16
16
17
17
import collections
18
18
19
- # Dependency imports
20
19
from absl import logging
21
20
import numpy as np
22
21
Original file line number Diff line number Diff line change 19
19
import inspect
20
20
import textwrap
21
21
22
- # Dependency imports
23
22
from absl import logging
24
23
import numpy as np
25
24
import six
Original file line number Diff line number Diff line change 14
14
# ============================================================================
15
15
"""Tests of the instruction language (and definitional interpreter)."""
16
16
17
- # Dependency imports
18
-
19
17
from tensorflow_probability .python .experimental .auto_batching import instructions
20
18
from tensorflow_probability .python .experimental .auto_batching import test_programs
21
19
from tensorflow_probability .python .internal import test_util as tfp_test_util
Original file line number Diff line number Diff line change 14
14
# ============================================================================
15
15
"""Tests of the FunctionCallOp lowering compilation pass."""
16
16
17
- # Dependency imports
18
-
19
17
from tensorflow_probability .python .experimental .auto_batching import allocation_strategy
20
18
from tensorflow_probability .python .experimental .auto_batching import lowering
21
19
from tensorflow_probability .python .experimental .auto_batching import numpy_backend
Original file line number Diff line number Diff line change 23
23
24
24
import collections
25
25
26
- # Dependency imports
27
26
import numpy as np
28
27
29
28
from tensorflow_probability .python .experimental .auto_batching import instructions
Original file line number Diff line number Diff line change 14
14
# ============================================================================
15
15
"""Tests for implementations of batched variables."""
16
16
17
- # Dependency imports
18
17
import hypothesis as hp
19
18
from hypothesis import strategies as hps
20
19
from hypothesis .extra import numpy as hpnp
Original file line number Diff line number Diff line change 12
12
# See the License for the specific language governing permissions and
13
13
# limitations under the License.
14
14
# ============================================================================
15
- # Dependency imports
16
-
17
15
from tensorflow_probability .python .experimental .auto_batching import stack_optimization as stack
18
16
from tensorflow_probability .python .experimental .auto_batching import test_programs
19
17
from tensorflow_probability .python .internal import test_util as tfp_test_util
Original file line number Diff line number Diff line change 22
22
import functools
23
23
import heapq
24
24
25
- # Dependency imports
26
25
from absl import logging
27
26
import six
28
27
from tensorflow_probability .python .experimental .auto_batching import instructions as inst
Original file line number Diff line number Diff line change 14
14
# ============================================================================
15
15
"""Tests of the FunctionCallOp lowering compilation pass."""
16
16
17
- # Dependency imports
18
-
19
17
import tensorflow .compat .v1 as tf1
20
18
21
19
from tensorflow_probability .python .experimental .auto_batching import allocation_strategy
Original file line number Diff line number Diff line change 14
14
# ============================================================================
15
15
"""Tests of the instruction language (and definitional interpreter)."""
16
16
17
- # Dependency imports
18
17
import numpy as np
19
18
20
19
from tensorflow_probability .python .experimental .auto_batching import instructions
Original file line number Diff line number Diff line change 20
20
import collections
21
21
import contextlib
22
22
23
- # Dependency imports
24
23
import six
25
24
import tensorflow .compat .v1 as tf1
26
25
import tensorflow .compat .v2 as tf
Original file line number Diff line number Diff line change 14
14
# ============================================================================
15
15
"""Tests for the TF implementations of auto-batched VM variables."""
16
16
17
- # Dependency imports
18
-
19
17
import hypothesis as hp
20
18
from hypothesis import strategies as hps
21
19
from hypothesis .extra import numpy as hpnp
Original file line number Diff line number Diff line change 16
16
17
17
import functools
18
18
19
- # Dependency imports
20
-
21
19
from absl .testing import parameterized
22
20
23
21
import numpy as np
Original file line number Diff line number Diff line change 17
17
import contextlib
18
18
import functools
19
19
20
- # Dependency imports
21
-
22
20
from absl import logging
23
21
import six
24
22
Original file line number Diff line number Diff line change 16
16
17
17
import functools
18
18
19
- # Dependency imports
20
-
21
19
from absl import flags
22
20
23
21
import numpy as np
You can’t perform that action at this time.
0 commit comments