FazBrowse GitHub Viewer
|
Trending
|
URL:
|
Home
Tools:
[Download Repo ZIP]
[View Raw Code]
[Original HTTPS Page]
TensorFlow.NET/src/TensorFlowNET.Core/Graphs/AutoGraph.cs at master · feelsyt/TensorFlow.NET · GitHub
feelsyt
/
TensorFlow.NET
Public
forked from
SciSharp/TensorFlow.NET
Notifications
You must be signed in to change notification settings
Fork
0
Star
0
Code
Pull requests
0
Actions
Projects
Security and quality
0
Insights
Additional navigation options
Code
Pull requests
Actions
Projects
Security and quality
Insights
Expand file tree
Breadcrumbs
TensorFlow.NET
/
src
/
TensorFlowNET.Core
/
Graphs
/
AutoGraph.cs
Copy path
More file actions
More file actions
Latest commit
History
History
History
83 lines (72 loc) · 2.65 KB
Breadcrumbs
TensorFlow.NET
/
src
/
TensorFlowNET.Core
/
Graphs
/
AutoGraph.cs
Copy path
File metadata and controls
83 lines (72 loc) · 2.65 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
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
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
using
System
;
using
System
.
Diagnostics
;
using
System
.
Linq
;
using
static
Tensorflow
.
Binding
;
namespace
Tensorflow
.
Graphs
{
public
class
AutoGraph
{
public
Func
<
Tensor
,
Tensor
>
to_graph
(
Func
<
Tensor
,
Tensor
>
func
,
TF_DataType
dtype
=
TF_DataType
.
TF_INT32
)
{
string
func_name
=
$
"
{
func
.
Method
.
Name
}
_
{
ops
.
uid_function
(
)
}
"
;
var
graph
=
new
FuncGraph
(
func_name
)
;
graph
.
as_default
(
)
;
var
input
=
tf
.
placeholder
(
dtype
)
;
var
output
=
func
(
input
)
;
var
opers
=
graph
.
_nodes_by_name
.
Values
.
Select
(
x
=>
x
as
Operation
)
.
ToArray
(
)
;
graph
.
ToGraph
(
opers
,
new
[
]
{
input
}
,
new
[
]
{
output
}
,
null
)
;
graph
.
Exit
(
)
;
return
(
Tensor
input
)
=>
{
if
(
tf
.
executing_eagerly
(
)
)
{
var
result
=
tf
.
Runner
.
TFE_Execute
(
tf
.
Context
,
tf
.
Context
.
DeviceName
,
func_name
,
new
[
]
{
input
}
,
null
,
1
)
;
return
result
[
0
]
;
}
var
s
=
tf
.
Session
(
input
.
graph
)
;
var
output
=
func
(
input
)
;
return
output
;
}
;
}
public
Func
<
Tensor
,
Tensor
,
Tensor
>
to_graph
(
Func
<
Tensor
,
Tensor
,
Tensor
>
func
,
params
TF_DataType
[
]
dtypes
)
{
string
func_name
=
$
"
{
func
.
Method
.
Name
}
_
{
ops
.
uid_function
(
)
}
"
;
var
graph
=
new
FuncGraph
(
func_name
)
;
graph
.
as_default
(
)
;
var
input1
=
tf
.
placeholder
(
dtypes
.
Length
>=
1
?
dtypes
[
0
]
:
tf
.
int32
)
;
var
input2
=
tf
.
placeholder
(
dtypes
.
Length
>=
2
?
dtypes
[
1
]
:
tf
.
int32
)
;
var
output
=
func
(
input1
,
input2
)
;
var
opers
=
graph
.
_nodes_by_name
.
Values
.
Select
(
x
=>
x
as
Operation
)
.
ToArray
(
)
;
graph
.
ToGraph
(
opers
,
new
[
]
{
input1
,
input2
}
,
new
[
]
{
output
}
,
null
)
;
graph
.
Exit
(
)
;
return
(
Tensor
a
,
Tensor
b
)
=>
{
if
(
tf
.
executing_eagerly
(
)
)
{
var
result
=
tf
.
Runner
.
TFE_Execute
(
tf
.
Context
,
tf
.
Context
.
DeviceName
,
func_name
,
new
[
]
{
a
,
b
}
,
null
,
1
)
;
return
result
[
0
]
;
}
var
s
=
tf
.
Session
(
a
.
graph
)
;
Debug
.
Assert
(
a
.
graph
==
b
.
graph
)
;
var
output
=
func
(
a
,
b
)
;
return
output
;
}
;
}
}
}
Back
|
FazBrowse Home
|
New Git URL