Home
last modified time | relevance | path

Searched refs:callable_options (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc445 feed_rewrites.reserve(options.callable_options.feed_size()); in PruneGraph()
447 fetch_rewrites.reserve(options.callable_options.fetch_size()); in PruneGraph()
450 for (int i = 0; i < options.callable_options.feed_size(); ++i) { in PruneGraph()
453 const string& feed = options.callable_options.feed(i); in PruneGraph()
456 options.callable_options.feed_devices(), in PruneGraph()
462 if (!options.callable_options.fetch_devices().empty() && in PruneGraph()
463 !options.callable_options.fetch_skip_sync()) { in PruneGraph()
470 for (int i = 0; i < options.callable_options.fetch_size(); ++i) { in PruneGraph()
473 const string& fetch = options.callable_options.fetch(i); in PruneGraph()
476 options.callable_options.fetch_devices(), in PruneGraph()
[all …]
Ddirect_session.cc433 const CallableOptions& callable_options, int64 global_step, in CreateDebuggerState() argument
437 callable_options.run_options().debug_options(), debugger_state)); in CreateDebuggerState()
438 std::vector<string> input_names(callable_options.feed().begin(), in CreateDebuggerState()
439 callable_options.feed().end()); in CreateDebuggerState()
440 std::vector<string> output_names(callable_options.fetch().begin(), in CreateDebuggerState()
441 callable_options.fetch().end()); in CreateDebuggerState()
442 std::vector<string> target_names(callable_options.target().begin(), in CreateDebuggerState()
443 callable_options.target().end()); in CreateDebuggerState()
475 CreateDebuggerState(executors_and_keys->callable_options, in RunInternal()
682 {executors_and_keys->callable_options.fetch().begin(), in RunInternal()
[all …]
Ddirect_session_test.cc200 CallableOptions callable_options; in TEST_F() local
201 TensorConnection* c = callable_options.add_tensor_connection(); in TEST_F()
204 callable_options.add_fetch(y_neg_ + ":0"); in TEST_F()
207 TF_ASSERT_OK(session->MakeCallable(callable_options, &handle)); in TEST_F()
223 CallableOptions callable_options; in TEST_F() local
224 TensorConnection* c = callable_options.add_tensor_connection(); in TEST_F()
227 callable_options.add_fetch(a_ + ":0"); in TEST_F()
228 callable_options.add_fetch(y_neg_ + ":0"); in TEST_F()
231 TF_ASSERT_OK(session->MakeCallable(callable_options, &handle)); in TEST_F()
254 CallableOptions callable_options; in TEST_F() local
[all …]
Dbuild_graph_options.cc24 for (auto& s : callable_options.feed()) { in DebugString()
28 for (auto& s : callable_options.fetch()) { in DebugString()
32 for (auto& s : callable_options.target()) { in DebugString()
Dbuild_graph_options.h28 CallableOptions callable_options; member
Ddirect_session.h111 ::tensorflow::Status MakeCallable(const CallableOptions& callable_options,
155 CallableOptions callable_options; member
230 const CallableOptions& callable_options,
/external/tensorflow/tensorflow/python/debug/wrappers/
Dframework.py424 callable_options=None): argument
448 if callable_runner and callable_options:
457 elif callable_options and (fetches or feed_dict):
471 elif callable_options:
474 callable_options)(*callable_runner_args)
492 if callable_options:
493 callable_options_id = id(callable_options)
497 new_callable_options.CopyFrom(callable_options)
524 elif callable_options:
643 def _make_callable_from_options(self, callable_options): argument
[all …]
Dlocal_cli_wrapper_test.py415 callable_options = config_pb2.CallableOptions()
416 callable_options.fetch.append("callable_b")
417 sess_callable = wrapped_sess._make_callable_from_options(callable_options)
438 callable_options = config_pb2.CallableOptions()
439 callable_options.feed.append("callable_ph1")
440 callable_options.fetch.append("callable_b")
441 sess_callable = wrapped_sess._make_callable_from_options(callable_options)
464 callable_options = config_pb2.CallableOptions()
465 callable_options.feed.append("callable_ph1")
466 callable_options.feed.append("callable_ph2")
[all …]
Dgrpc_wrapper.py219 callable_options=None): argument
231 callable_options=callable_options)
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.cc118 CallableOptions callable_options; in RunOnce() local
120 *callable_options.mutable_run_options() = run_options; in RunOnce()
124 callable_options.add_feed(name); in RunOnce()
128 callable_options.add_fetch(output_tensor_name); in RunOnce()
131 callable_options.add_target(target_node_name); in RunOnce()
135 TF_RETURN_IF_ERROR(session->MakeCallable(callable_options, &callable_handle)); in RunOnce()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc174 const TF_Buffer* callable_options, int64_t* out_handle, in MakeCallableHelper() argument
177 if (callable_options != nullptr && in MakeCallableHelper()
178 !callable_options_proto.ParseFromArray(callable_options->data, in MakeCallableHelper()
179 callable_options->length)) { in MakeCallableHelper()
196 const TF_Buffer* callable_options, in TF_DeprecatedSessionMakeCallable() argument
199 MakeCallableHelper(session->session, callable_options, out_handle, in TF_DeprecatedSessionMakeCallable()
203 const TF_Buffer* callable_options, in TF_SessionMakeCallable() argument
205 MakeCallableHelper(session->session, callable_options, out_handle, in TF_SessionMakeCallable()
Dsession_ref.cc310 const CallableOptions& callable_options, in RecordMakeCallable() argument
315 *req->mutable_options() = callable_options; in RecordMakeCallable()
317 RUN_WITH_TIMESTAMP(MakeCallable, callable_options, handle); in RecordMakeCallable()
463 Status SessionRef::MakeCallable(const CallableOptions& callable_options, in MakeCallable() argument
465 LOG_AND_RUN_OPERATION(MakeCallable, callable_options, out_handle); in MakeCallable()
Dtf_session_helper.h67 const TF_Buffer* callable_options,
71 const TF_Buffer* callable_options,
Dsession_ref.h67 Status MakeCallable(const CallableOptions& callable_options,
Dsession.py1418 def __init__(self, session, callable_options): argument
1422 compat.as_bytes(callable_options.SerializeToString()))
1461 def _make_callable_from_options(self, callable_options): argument
1472 return BaseSession._Callable(self, callable_options)
/external/tensorflow/tensorflow/cc/client/
Dclient_session.cc130 Status ClientSession::MakeCallable(const CallableOptions& callable_options, in MakeCallable() argument
133 return impl()->session_->MakeCallable(callable_options, out_handle); in MakeCallable()
Dclient_session.h97 Status MakeCallable(const CallableOptions& callable_options,
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_session.cc78 callable_opts_(bopts.callable_options), in ReffedClientGraph()
108 const CallableOptions& callable_options() { return callable_opts_; } in callable_options() function in tensorflow::MasterSession::ReffedClientGraph
1086 CallableOptions* callable_opts = &opts->callable_options; in BuildBuildGraphOptions()
1113 CallableOptions* callable_opts = &opts->callable_options; in BuildBuildGraphOptions()
1129 for (const string& name : opts.callable_options.feed()) { in HashBuildGraphOptions()
1132 for (const string& name : opts.callable_options.target()) { in HashBuildGraphOptions()
1135 for (const string& name : opts.callable_options.fetch()) { in HashBuildGraphOptions()
1140 opts.callable_options.run_options().debug_options(); in HashBuildGraphOptions()
1152 for (const string& name : opts.callable_options.feed()) { in BuildGraphOptionsString()
1156 for (const string& name : opts.callable_options.target()) { in BuildGraphOptionsString()
[all …]
/external/tensorflow/tensorflow/core/public/
Dsession.h205 virtual Status MakeCallable(const CallableOptions& callable_options, in MakeCallable() argument
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.h98 Status MakeCallable(const CallableOptions& callable_options,
Dgrpc_session.cc408 Status GrpcSession::MakeCallable(const CallableOptions& callable_options, in MakeCallable() argument
412 *req.mutable_options() = callable_options; in MakeCallable()