Home
last modified time | relevance | path

Searched refs:add_fetch (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util_test.cc58 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()
62 fetch = config.add_fetch(); in TEST()
98 config.add_fetch(); in TEST()
106 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()
127 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()
130 fetch = config.add_fetch(); in TEST()
151 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()
154 fetch = config.add_fetch(); in TEST()
163 auto* fetch = config.add_fetch(); in FetchesConfig()
Dtf2xla_test.cc65 config.add_fetch()->mutable_id()->set_node_name("sum"); in SumConfig()
Dgraph_compiler_util.cc288 auto ph = prune_config.add_fetch(); in InitGraph()
Dxla_jit_compiled_cpu_function_test.cc80 tf2xla::Fetch* sum = config.add_fetch(); in SumConfig()
/external/tensorflow/tensorflow/python/client/
Dsession_ref.cc152 req->add_fetch(name); in RecordRun()
266 req->add_fetch(output); in RecordPRunSetup()
295 req->add_fetch(output); in RecordPRun()
345 tensor.AsProtoTensorContent(resp->add_fetch()); in RecordRunCallable()
/external/tensorflow/tensorflow/cc/client/
Dclient_session_test.cc195 options.add_fetch(c.node()->name()); in TEST()
231 options.add_fetch(c.node()->name()); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_test.cc74 ret.add_fetch(fetch); in MakeCallableOptions()
352 callable_options.add_fetch(y_neg_ + ":0"); in TEST_F()
375 callable_options.add_fetch(a_ + ":0"); in TEST_F()
376 callable_options.add_fetch(y_neg_ + ":0"); in TEST_F()
406 callable_options.add_fetch(y_ + ":0"); in TEST_F()
420 callable_options.add_fetch(y_ + ":0"); in TEST_F()
435 callable_options.add_fetch(y_ + ":0"); in TEST_F()
449 callable_options.add_fetch(y_ + ":0"); in TEST_F()
467 callable_options.add_fetch(z_ + ":0"); in TEST_F()
482 callable_options.add_fetch(y_neg_ + ":0"); in TEST_F()
[all …]
Ddirect_session.cc1543 callable_options.add_fetch(output); in GetOrCreateExecutors()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers_test.cc45 request->add_fetch("fetch_x:0"); in BuildRunStepRequest()
46 request->add_fetch("fetch_y:0"); in BuildRunStepRequest()
Dmessage_wrappers.cc78 void InMemoryRunStepRequest::add_fetch(const string& name) { in add_fetch() function in tensorflow::InMemoryRunStepRequest
122 proto_version_->add_fetch(fetch_name(i)); in ToProto()
182 void MutableProtoRunStepRequest::add_fetch(const string& name) { in add_fetch() function in tensorflow::MutableProtoRunStepRequest
183 request_.add_fetch(name); in add_fetch()
Dmessage_wrappers.h110 virtual void add_fetch(const string& name) = 0;
141 void add_fetch(const string& name) override;
192 void add_fetch(const string& name) override;
Dmaster_test.cc110 req.add_fetch(fetch_name); in RunStep()
/external/tensorflow/tensorflow/compiler/aot/
Dcodegen_test.cc198 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.cc235 req->add_fetch(name); in RunHelper()
328 req.add_fetch(fetch); in PRunSetup()
Dgrpc_session_test.cc151 opts.add_fetch(node_names[2] + ":0"); in TEST()
190 opts.add_fetch(fetch); in TEST()
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.cc167 callable_options.add_fetch(output_tensor_name); in RunOnce()