| /external/tensorflow/tensorflow/examples/image_retraining/ |
| retrain_test.py | 64 with tf.Session() as sess:
66 self.assertIsNotNone(sess.graph.get_tensor_by_name('DistortJPGInput:0'))
67 self.assertIsNotNone(sess.graph.get_tensor_by_name('DistortResult:0'))
72 with tf.Session() as sess:
76 self.assertIsNotNone(sess.graph.get_tensor_by_name('final:0'))
81 with tf.Session() as sess:
85 self.assertIsNotNone(sess.graph.get_tensor_by_name('final:0'))
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| padding_fifo_queue_test.py | 121 with self.test_session() as sess:
129 sess.run(enqueue_op)
147 with self.test_session() as sess:
161 results.append(sess.run(dequeued_t))
185 with self.test_session() as sess:
196 sess.run(enqueue_op)
202 results.append(sess.run(dequeued_t))
215 with self.test_session() as sess:
227 x_val, y_val = sess.run(dequeued_t)
317 with self.test_session() as sess
[all...] |
| variables_test.py | 145 with self.test_session() as sess:
151 sess.run(variables.local_variables_initializer())
154 self.assertEqual([], list(sess.run(copy_op)))
202 with self.test_session() as sess:
223 sess.run(v1.initializer)
224 self.assertEqual([1], sess.run(v1))
225 sess.run(v2.initializer)
226 self.assertEqual([2], sess.run(v2))
229 sess.run(v0)
232 sess.run(add
[all...] |
| sparse_conditional_accumulator_test.py | 55 def _assertEqual_nparray(self, expected_array, result, sess):
116 with self.test_session() as sess:
133 result = sess.run(q.take_indexed_slices_grad(1))
135 self._assertEqual_nparray(sum_elems / len(elems), result, sess)
138 with self.test_session() as sess:
164 result = sess.run(accums[i].take_indexed_slices_grad(1))
168 with self.test_session() as sess:
182 val = sess.run(takeg_t)
188 with self.test_session() as sess:
203 val = sess.run(takeg_t
[all...] |
| summary_ops_test.py | 37 with self.test_session() as sess:
40 value = sess.run(summ)
48 with self.test_session() as sess:
51 value = sess.run(summ)
59 with self.test_session() as sess:
64 value = sess.run(merge)
|
| /external/tensorflow/tensorflow/contrib/opt/python/training/ |
| moving_average_optimizer_test.py | 49 with self.test_session(graph=ops.Graph()) as sess:
104 train_saver.save(sess, save_path)
105 train_saver.restore(sess, save_path)
114 train_saver_subset.save(sess, subset_save_path)
121 train_saver_subset.restore(sess, subset_save_path)
127 train_saver.restore(sess, save_path)
137 train_saver.save(sess, save_path)
138 inference_saver.restore(sess, save_path)
146 train_saver.restore(sess, save_path)
157 train_saver.save(sess, save_path
[all...] |
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| core_rnn_cell_test.py | 50 with self.test_session() as sess:
55 sess.run([variables_lib.global_variables_initializer()])
56 res = sess.run([l], {x.name: np.array([[1., 2.]])})
71 with self.test_session() as sess:
83 sess.run([variables_lib.global_variables_initializer()])
84 res = sess.run([g], {
91 with self.test_session() as sess:
110 sess.run([variables_lib.global_variables_initializer()])
111 res = sess.run([g], {
118 with self.test_session() as sess
[all...] |
| gru_ops_test.py | 44 with self.test_session(use_gpu=True, graph=ops.Graph()) as sess:
55 sess.run(variables.global_variables_initializer())
58 sess.run(output, feed)
61 with self.test_session(use_gpu=True, graph=ops.Graph()) as sess:
80 sess.run([variables.global_variables_initializer()])
81 basic_res = sess.run([output], {x: x_value, h: h_value})
86 sess.run([variables.global_variables_initializer()])
87 block_res = sess.run([output], {x: x_value, h: h_value})
94 with self.test_session(use_gpu=True, graph=ops.Graph()) as sess:
124 sess.run([variables.global_variables_initializer()]
[all...] |
| rnn_cell_test.py | 50 with self.test_session() as sess:
69 sess.run([variables.global_variables_initializer()])
70 res = sess.run(
84 with self.test_session() as sess:
101 sess.run([variables.global_variables_initializer()])
102 res = sess.run(
123 with self.test_session() as sess:
151 sess.run([variables.global_variables_initializer()])
152 res = sess.run([output, state])
169 with self.test_session() as sess
[all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| cudnn_rnn_test.py | 213 def Feed(self, sess, inputs, initial_state=None, return_sum=True):
219 return sess.run(
224 return sess.run(
315 with self.test_session(use_gpu=True) as sess:
316 sess.run(variables.global_variables_initializer())
317 (total_sum1_v, total_sum2_v, total_sum3_v) = sess.run(
366 with self.test_session(use_gpu=True, graph=g) as sess:
367 sess.run(variables.global_variables_initializer())
368 sess.run(train_op)
404 with self.test_session(use_gpu=True, graph=g) as sess
[all...] |
| /external/tensorflow/tensorflow/contrib/kfac/python/kernel_tests/ |
| utils_test.py | 130 with ops.Graph().as_default(), self.test_session() as sess:
136 sess.run(output), np.array([[1., 2.], [4., 3.], [1., 2.]]))
139 with ops.Graph().as_default(), self.test_session() as sess:
144 self.assertAllClose(sess.run(output), np.array([[1., 2.], [4., 3.]]))
154 with ops.Graph().as_default(), self.test_session() as sess:
159 ans_tf = sess.run(utils.kronecker_product(mat1_tf, mat2_tf))
164 with ops.Graph().as_default(), self.test_session() as sess:
169 output = sess.run(utils.mat2d_to_layer_params(vector_template, mat2d))
178 with ops.Graph().as_default(), self.test_session() as sess:
183 output = sess.run(utils.mat2d_to_layer_params(vector_template, mat2d)
[all...] |
| estimator_test.py | 130 with self._graph.as_default(), self.test_session() as sess:
149 sess.run(variables.global_variables_initializer())
150 initial_cov_values = sess.run(cov_matrices)
160 new_cov_values = sess.run(cov_matrices)
172 sess.run(cov_update_op)
173 sess.run(increment_global_step)
177 with self._graph.as_default(), self.test_session() as sess:
197 sess.run(variables.global_variables_initializer())
198 initial_inv_values = sess.run(inv_matrices)
214 sess.run(
[all...] |
| /external/tensorflow/tensorflow/python/client/ |
| session_test.py | 186 with session.Session() as sess:
191 with sess.graph._original_op(a.op):
193 with sess.graph._original_op(b.op):
217 with session.Session() as sess:
219 res = sess.run(a)
221 res = sess.run(a.op) # An op, not a tensor.
223 tensor_runner = sess.make_callable(a)
226 op_runner = sess.make_callable(a.op)
231 with session.Session() as sess:
233 res = sess.run(a.name
[all...] |
| session_list_devices_test.py | 38 with session.Session() as sess:
39 devices = sess.list_devices()
63 with session.Session(server.target) as sess:
64 devices = sess.list_devices()
78 with session.Session(server1.target, config=config) as sess:
79 devices = sess.list_devices()
|
| /external/tensorflow/tensorflow/python/training/ |
| queue_runner_test.py | 44 with self.test_session() as sess:
52 threads = qr.create_threads(sess)
64 with self.test_session() as sess:
73 threads = qr.create_threads(sess)
87 with self.test_session() as sess:
91 threads = qr.create_threads(sess)
103 with self.test_session() as sess:
111 threads = qr.create_threads(sess)
131 with self.test_session() as sess:
143 threads = qr.create_threads(sess, coord
[all...] |
| ftrl_test.py | 40 with self.test_session() as sess:
57 v0_val, v1_val = sess.run([var0, var1])
65 v0_val, v1_val = sess.run([var0, var1])
79 with self.test_session() as sess:
93 v0_val, v1_val = sess.run([var0, var1])
100 v0_val, v1_val = sess.run([var0, var1])
125 with self.test_session() as sess:
139 v0_val, v1_val = sess.run([var0, var1])
146 v0_val, v1_val = sess.run([var0, var1])
154 with self.test_session() as sess
[all...] |
| /external/tensorflow/tensorflow/contrib/distributions/python/kernel_tests/bijectors/ |
| affine_test.py | 43 with self.test_session() as sess:
51 return sess.run(fun(x), feed_dict={x: x_value})
66 with self.test_session() as sess:
74 return sess.run(fun(x), feed_dict={x: x_value})
88 with self.test_session() as sess:
96 return sess.run(fun(x), feed_dict={x: x_value})
115 with self.test_session() as sess:
123 return sess.run(fun(x), feed_dict={x: x_value})
137 with self.test_session() as sess:
145 return sess.run(fun(x), feed_dict={x: x_value}
[all...] |
| real_nvp_test.py | 46 with self.test_session() as sess:
64 ] = sess.run([
77 with self.test_session() as sess:
88 sess_run_fn=sess.run,
97 with self.test_session() as sess:
108 sess_run_fn=sess.run,
|
| /external/tensorflow/tensorflow/contrib/slim/python/slim/nets/ |
| inception_v1_test.py | 146 with self.test_session() as sess:
156 pre_pool_out = sess.run(pre_pool, feed_dict=feed_dict)
170 with self.test_session() as sess:
171 sess.run(variables.global_variables_initializer())
172 output = sess.run(logits, {inputs: images.eval()})
185 with self.test_session() as sess:
186 sess.run(variables.global_variables_initializer())
187 output = sess.run(predictions)
203 with self.test_session() as sess:
204 sess.run(variables.global_variables_initializer()
[all...] |
| /external/tensorflow/tensorflow/python/tools/ |
| optimize_for_inference_test.py | 133 with self.test_session() as sess:
160 original_graph_def = sess.graph_def
161 original_result = sess.run(["output:0"])
165 with self.test_session() as sess:
168 optimized_result = sess.run(["optimized/output:0"])
177 with self.test_session(use_gpu=use_gpu) as sess:
211 original_graph_def = sess.graph_def
212 original_result = sess.run(["output:0"])
216 with self.test_session(use_gpu=use_gpu) as sess:
219 optimized_result = sess.run(["optimized/output:0"]
[all...] |
| /external/tensorflow/tensorflow/compiler/tests/ |
| stateless_random_ops_test.py | 57 with self.test_session() as sess, self.test_scope():
62 y = sess.run(x, {seed_t: [0x12345678, 0xabcdef12]})
76 with self.test_session() as sess, self.test_scope():
82 y = sess.run(x, {seed_t: [565656, 121212]})
104 with self.test_session() as sess, self.test_scope():
110 y = sess.run(x, {seed_t: [25252, 314159]})
|
| /external/tensorflow/tensorflow/contrib/boosted_trees/python/kernel_tests/ |
| stats_accumulator_ops_test.py | 32 with self.test_session() as sess:
49 num_updates, partition, bucket_ids, grads, hessians = sess.run(
60 with self.test_session() as sess:
77 num_updates, partition, bucket_ids, grads, hessians = sess.run(
89 with self.test_session() as sess:
111 num_updates, partition, feature, grads, hessians = sess.run(
121 with self.test_session() as sess:
142 num_updates_2, partition_2, feature_2, grads_2, hessians_2) = sess.run(
162 with self.test_session() as sess:
188 num_updates, partition, feature, grads, hessians = sess.run
[all...] |
| /external/tensorflow/tensorflow/contrib/gan/python/eval/python/ |
| sliced_wasserstein_test.py | 68 with self.test_session() as sess:
69 pyramid_tf = sess.run(
82 with self.test_session() as sess:
83 wscores = [sess.run(x) for x in wfunc]
98 with self.test_session() as sess:
99 wscores = [sess.run(x) for x in wfunc]
|
| /external/tensorflow/tensorflow/contrib/lite/toco/python/ |
| toco_from_protos_test.py | 37 def _run(self, sess, in_tensor, out_tensor, should_succeed):
41 sess: Active TensorFlow session containing graph.
47 graph_def = sess.graph_def
82 with tf.Session() as sess:
88 self._run(sess, img, out, True)
91 self._run(sess, img, out2, True)
93 self._run(sess, img, img, False)
|
| /external/tensorflow/tensorflow/go/op/ |
| op_test.go | 71 sess, err := tf.NewSession(graph, nil)
84 fetched, err := sess.Run(feeds, []tf.Output{z}, nil)
115 sess, err := tf.NewSession(graph, nil)
119 if _, err := sess.Run(nil, nil, []*tf.Operation{init}); err != nil {
122 results, err := sess.Run(nil, next, nil)
130 if _, err := sess.Run(nil, next, nil); err == nil {
131 t.Errorf("Expected sess.Run() to fail since the iterator should have reached the end of the dataset")
|
