Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/debug/wrappers/
Dframework.py404 callable_options=None): argument
428 if callable_runner and callable_options:
437 elif callable_options and (fetches or feed_dict):
463 elif callable_options:
466 callable_options)(*callable_runner_args)
484 callable_runner, callable_runner_args, callable_options)
488 callable_runner, callable_runner_args, callable_options)
493 elif callable_options:
496 callable_options)
527 callable_options): argument
[all …]
Dlocal_cli_wrapper_test.py423 callable_options = config_pb2.CallableOptions()
424 callable_options.fetch.append("callable_b")
425 sess_callable = wrapped_sess._make_callable_from_options(callable_options)
446 callable_options = config_pb2.CallableOptions()
447 callable_options.feed.append("callable_ph1")
448 callable_options.fetch.append("callable_b")
449 sess_callable = wrapped_sess._make_callable_from_options(callable_options)
473 callable_options = config_pb2.CallableOptions()
474 callable_options.feed.append("callable_ph1")
475 callable_options.feed.append("callable_ph2")
[all …]
Dgrpc_wrapper.py219 callable_options=None): argument
231 callable_options=callable_options)
/external/tensorflow/tensorflow/core/common_runtime/
Dgraph_execution_state.cc496 feed_rewrites.reserve(options.callable_options.feed_size()); in PruneGraph()
498 fetch_rewrites.reserve(options.callable_options.fetch_size()); in PruneGraph()
501 for (int i = 0; i < options.callable_options.feed_size(); ++i) { in PruneGraph()
504 const string& feed = options.callable_options.feed(i); in PruneGraph()
507 options.callable_options.feed_devices(), in PruneGraph()
513 if (!options.callable_options.fetch_devices().empty() && in PruneGraph()
514 !options.callable_options.fetch_skip_sync()) { in PruneGraph()
521 for (int i = 0; i < options.callable_options.fetch_size(); ++i) { in PruneGraph()
524 const string& fetch = options.callable_options.fetch(i); in PruneGraph()
527 options.callable_options.fetch_devices(), in PruneGraph()
[all …]
Ddirect_session.cc463 const CallableOptions& callable_options, int64 global_step, in CreateDebuggerState() argument
467 callable_options.run_options().debug_options(), debugger_state)); in CreateDebuggerState()
468 std::vector<string> input_names(callable_options.feed().begin(), in CreateDebuggerState()
469 callable_options.feed().end()); in CreateDebuggerState()
470 std::vector<string> output_names(callable_options.fetch().begin(), in CreateDebuggerState()
471 callable_options.fetch().end()); in CreateDebuggerState()
472 std::vector<string> target_names(callable_options.target().begin(), in CreateDebuggerState()
473 callable_options.target().end()); in CreateDebuggerState()
519 CreateDebuggerState(executors_and_keys->callable_options, in RunInternal()
753 {executors_and_keys->callable_options.fetch().begin(), in RunInternal()
[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()
Ddirect_session_test.cc348 CallableOptions callable_options; in TEST_F() local
349 TensorConnection* c = callable_options.add_tensor_connection(); in TEST_F()
352 callable_options.add_fetch(y_neg_ + ":0"); in TEST_F()
355 TF_ASSERT_OK(session->MakeCallable(callable_options, &handle)); in TEST_F()
371 CallableOptions callable_options; in TEST_F() local
372 TensorConnection* c = callable_options.add_tensor_connection(); in TEST_F()
375 callable_options.add_fetch(a_ + ":0"); in TEST_F()
376 callable_options.add_fetch(y_neg_ + ":0"); in TEST_F()
379 TF_ASSERT_OK(session->MakeCallable(callable_options, &handle)); in TEST_F()
402 CallableOptions callable_options; in TEST_F() local
[all …]
Dbuild_graph_options.h28 CallableOptions callable_options; member
Ddirect_session.h121 ::tensorflow::Status MakeCallable(const CallableOptions& callable_options,
176 CallableOptions callable_options; member
251 const CallableOptions& callable_options,
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.cc119 CallableOptions callable_options; in RunOnce() local
121 *callable_options.mutable_run_options() = run_options; in RunOnce()
125 callable_options.add_feed(name); in RunOnce()
129 callable_options.add_fetch(output_tensor_name); in RunOnce()
132 callable_options.add_target(target_node_name); in RunOnce()
136 TF_RETURN_IF_ERROR(session->MakeCallable(callable_options, &callable_handle)); in RunOnce()
427 Status MakeCallable(const CallableOptions& callable_options, in MakeCallable() argument
429 return wrapped_->MakeCallable(callable_options, out_handle); in MakeCallable()
/external/tensorflow/tensorflow/python/client/
Dtf_session_helper.cc175 const TF_Buffer* callable_options, int64_t* out_handle, in MakeCallableHelper() argument
178 if (callable_options != nullptr && in MakeCallableHelper()
179 !callable_options_proto.ParseFromArray(callable_options->data, in MakeCallableHelper()
180 callable_options->length)) { in MakeCallableHelper()
197 const TF_Buffer* callable_options, in TF_DeprecatedSessionMakeCallable() argument
199 MakeCallableHelper(session->session, callable_options, out_handle, status); in TF_DeprecatedSessionMakeCallable()
202 const TF_Buffer* callable_options, in TF_SessionMakeCallable() argument
204 MakeCallableHelper(session->session, callable_options, out_handle, status); 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,
70 const TF_Buffer* callable_options,
Dsession_ref.h67 Status MakeCallable(const CallableOptions& callable_options,
Dsession.py1455 def __init__(self, session, callable_options): argument
1459 compat.as_bytes(callable_options.SerializeToString()))
1496 def _make_callable_from_options(self, callable_options): argument
1507 return BaseSession._Callable(self, callable_options)
Dtf_session_wrapper.cc153 [](TF_Session* session, const TF_Buffer* callable_options) { in PYBIND11_MODULE() argument
160 tensorflow::TF_SessionMakeCallable(session, callable_options, in PYBIND11_MODULE()
/external/tensorflow/tensorflow/cc/client/
Dclient_session.cc157 Status ClientSession::MakeCallable(const CallableOptions& callable_options, in MakeCallable() argument
160 return impl()->session_->MakeCallable(callable_options, out_handle); in MakeCallable()
Dclient_session.h111 Status MakeCallable(const CallableOptions& callable_options,
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmaster_session.cc80 callable_opts_(bopts.callable_options), in ReffedClientGraph()
110 const CallableOptions& callable_options() { return callable_opts_; } in callable_options() function in tensorflow::MasterSession::ReffedClientGraph
1141 CallableOptions* callable_opts = &opts->callable_options; in BuildBuildGraphOptions()
1168 CallableOptions* callable_opts = &opts->callable_options; in BuildBuildGraphOptions()
1184 for (const string& name : opts.callable_options.feed()) { in HashBuildGraphOptions()
1187 for (const string& name : opts.callable_options.target()) { in HashBuildGraphOptions()
1190 for (const string& name : opts.callable_options.fetch()) { in HashBuildGraphOptions()
1195 opts.callable_options.run_options().debug_options(); in HashBuildGraphOptions()
1207 for (const string& name : opts.callable_options.feed()) { in BuildGraphOptionsString()
1211 for (const string& name : opts.callable_options.target()) { in BuildGraphOptionsString()
[all …]
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.h102 Status MakeCallable(const CallableOptions& callable_options,
Dgrpc_session.cc425 Status GrpcSession::MakeCallable(const CallableOptions& callable_options, in MakeCallable() argument
429 *req.mutable_options() = callable_options; in MakeCallable()
/external/tensorflow/tensorflow/core/public/
Dsession.h239 virtual Status MakeCallable(const CallableOptions& callable_options, in MakeCallable() argument