-
Notifications
You must be signed in to change notification settings - Fork 3
/
part_exe.py
84 lines (79 loc) · 1.99 KB
/
part_exe.py
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
84
import tensorflow as tf
import numpy as np
import time
import sys
with tf.device('/cpu:0'):
a = tf.placeholder(tf.float32, shape=[])
b = tf.placeholder(tf.float32, shape=[])
c = tf.placeholder(tf.float32, shape=[])
r1 = tf.add(a, b)
r4 = tf.add(r1,r1)
r2 = tf.multiply(r1,r1)
r3 = tf.multiply(r4,r4)
with tf.Session() as sess:
s1 = time.time()
h = sess.partial_run_setup([r2, r3], [a,b])
s2 = time.time()
res = sess.partial_run(h, r2,feed_dict={a:2,b:2})
s3 = time.time()
print (res)
re = sess.partial_run(h,r3)
s4 = time.time()
print(re)
print (s2-s1)
print (s3-s2)
print (s4-s3)
sys.exit()
with tf.Session() as sess:
a = tf.placeholder(tf.float32, shape=[])
#a = tf.constant([0.01]*10000)
#b = tf.constant([0.02]*10000)
b = tf.placeholder(tf.float32, shape=[])
#c = tf.placeholder(tf.float32, shape=[])
r1 = tf.add(a, b)
#r1 = tf.tensordot(a,b,1)
r2 = tf.multiply(r1, r1)
r3 = tf.multiply(r1,r1)
s1=time.time()
#for i in range(100):
h = sess.partial_run_setup([r1,r2, r3],[a,b])
#s2=time.time()
res = sess.partial_run(h,r1,feed_dict={a:1,b:2})
print (res)
res = sess.partial_run(h, r2)
print (res)
#s3=time.time()
#print (res)
res = sess.partial_run(h, r3)
print (res)
s4 = time.time()
#for i in range(100):
#res = sess.run(r2)
#s5 = time.time()
#print (s2-s1)
#print (s3-s2)
#print (s4-s1)
#print (s5-s4)
'''
s=time.time()
for i in range(10000):
sess.run(d)
#sess.run(y)
#res = sess.run(y)
duration = time.time() - s
print (duration)
s=time.time()
for i in range(10000):
#sess.run(d)
sess.run(y)
#res = sess.run(y)
duration = time.time() - s
print (duration)
s=time.time()
for i in range(10000):
sess.run(d)
sess.run(y)
#res = sess.run(y)
duration = time.time() - s
print (duration)
'''