Searched refs:add_fetch (Results 1 – 15 of 15) sorted by relevance
56 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()60 fetch = config.add_fetch(); in TEST()96 config.add_fetch(); in TEST()104 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()125 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()128 fetch = config.add_fetch(); in TEST()149 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()152 fetch = config.add_fetch(); in TEST()161 auto* fetch = config.add_fetch(); in FetchesConfig()
65 config.add_fetch()->mutable_id()->set_node_name("sum"); in SumConfig()
67 tf2xla::Fetch* sum = config.add_fetch(); in SumConfig()
151 req->add_fetch(name); in RecordRun()265 req->add_fetch(output); in RecordPRunSetup()294 req->add_fetch(output); in RecordPRun()344 tensor.AsProtoTensorContent(resp->add_fetch()); in RecordRunCallable()
68 ret.add_fetch(fetch); in MakeCallableOptions()204 callable_options.add_fetch(y_neg_ + ":0"); in TEST_F()227 callable_options.add_fetch(a_ + ":0"); in TEST_F()228 callable_options.add_fetch(y_neg_ + ":0"); in TEST_F()258 callable_options.add_fetch(y_ + ":0"); in TEST_F()273 callable_options.add_fetch(y_ + ":0"); in TEST_F()288 callable_options.add_fetch(y_ + ":0"); in TEST_F()302 callable_options.add_fetch(y_ + ":0"); in TEST_F()320 callable_options.add_fetch(z_ + ":0"); in TEST_F()335 callable_options.add_fetch(y_neg_ + ":0"); in TEST_F()[all …]
1426 callable_options.add_fetch(output); in GetOrCreateExecutors()
45 request->add_fetch("fetch_x:0"); in BuildRunStepRequest()46 request->add_fetch("fetch_y:0"); in BuildRunStepRequest()
80 void InMemoryRunStepRequest::add_fetch(const string& name) { in add_fetch() function in tensorflow::InMemoryRunStepRequest124 proto_version_->add_fetch(fetch_name(i)); in ToProto()184 void MutableProtoRunStepRequest::add_fetch(const string& name) { in add_fetch() function in tensorflow::MutableProtoRunStepRequest185 request_.add_fetch(name); in add_fetch()
107 virtual void add_fetch(const string& name) = 0;138 void add_fetch(const string& name) override;189 void add_fetch(const string& name) override;
109 req.add_fetch(fetch_name); in RunStep()
109 options.add_fetch(c.node()->name()); in TEST()
174 tf2xla::Fetch* fetch = config.add_fetch(); in TEST()
218 req->add_fetch(name); in RunHelper()311 req.add_fetch(fetch); in PRunSetup()
151 opts.add_fetch(node_names[2] + ":0"); in TEST()190 opts.add_fetch(fetch); in TEST()
128 callable_options.add_fetch(output_tensor_name); in RunOnce()