| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| partial_run_mgr.cc | 38 std::unique_ptr<PartialRunState> partial_run = MakeUnique<PartialRunState>(); local
39 partial_run->cancellation_manager = MakeUnique<CancellationManager>();
40 *cancellation_manager = partial_run->cancellation_manager.get();
41 step_id_to_partial_run_[step_id] = std::move(partial_run);
54 // If we found the partial_run, we call the final callback, if it
83 // If we found the partial_run, we set the final callback to call only
|
| /external/tensorflow/tensorflow/python/client/ |
| session_partial_run_test.py | 52 res = sess.partial_run(h, r1, feed_dict={a: 1, b: 2})
55 res = sess.partial_run(h, r2, feed_dict={c: temp})
60 res = sess.partial_run(h2, r1, feed_dict={a: 1, b: 2})
63 res = sess.partial_run(h2, r2, feed_dict={c: temp})
74 res = sess.partial_run(h, r1, feed_dict={a: 1, b: 2})
86 res = sess.partial_run(h1, r1, feed_dict={a: 1, b: 2})
89 res = sess.partial_run(h2, r1, feed_dict={a: temp, b: 9})
91 res = sess.partial_run(h2, r2, feed_dict={c: 7})
106 res = sess.partial_run(h, outputs[i], feed_dict={inputs[i]: 1.0})
122 r2 = sess.partial_run(h, [b, c]
[all...] |
| session_clusterspec_prop_test.py | 486 res = sess.partial_run(h, r1, feed_dict={a: 1, b: 2})
488 res = sess.partial_run(h, r2, feed_dict={c: 3})
|
| session.py | 62 def partial_run(self, handle, fetches, feed_dict=None): member in class:SessionInterface
64 raise NotImplementedError('partial_run')
916 def partial_run(self, handle, fetches, feed_dict=None): member in class:BaseSession
[all...] |
| session_test.py | 168 s.partial_run('foo', r1, feed_dict={a: 1, b: 2})
1664 partial_run = sess.partial_run_setup([squared_tensor], []) variable in class:SessionTest.testRegisterFetchAndFeedConversionFunctions.SquaredTensor
[all...] |
| /external/tensorflow/tensorflow/python/debug/wrappers/ |
| framework.py | 570 def partial_run(self, handle, fetches, feed_dict=None): member in class:BaseDebugWrapperSession
572 "partial_run is not implemented for debug-wrapper sessions.")
[all...] |
