tensorflow_stubs/stubs/tensorflow/graph_util.pyi at main · hmc-cs-mdrissi/tensorflow_stubs

History

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

from typing import Mapping, overload

from tensorflow import Operation, Tensor

from tensorflow.compat.v1 import GraphDef

@overload

def import_graph_def(

graph_def: GraphDef,

input_map: Mapping[str, Tensor] | None,

return_elements: list[str],

) -> list[Operation | Tensor]: ...

@overload

def import_graph_def(

graph_def: GraphDef,

*,

return_elements: list[str],

) -> list[Operation | Tensor]: ...

@overload

def import_graph_def(

graph_def: GraphDef,

input_map: Mapping[str, Tensor] | None = None,

return_elements: None = None,

) -> None: ...

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

graph_util.pyi

graph_util.pyi

Files

graph_util.pyi

Latest commit

History

graph_util.pyi

File metadata and controls