Home
last modified time | relevance | path

Searched refs:add_feed (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util_test.cc49 tf2xla::Feed* feed = config.add_feed(); in TEST()
53 feed = config.add_feed(); in TEST()
73 tf2xla::Feed* feed = config.add_feed(); in TEST()
80 config.add_feed(); in TEST()
86 tf2xla::Feed* feed = config.add_feed(); in TEST()
94 tf2xla::Feed* feed = config.add_feed(); in TEST()
102 tf2xla::Feed* feed = config.add_feed(); in TEST()
112 tf2xla::Feed* feed = config.add_feed(); in TEST()
115 feed = config.add_feed(); in TEST()
123 tf2xla::Feed* feed = config.add_feed(); in TEST()
[all …]
Dtf2xla_test.cc63 config.add_feed()->mutable_id()->set_node_name("x"); in SumConfig()
64 config.add_feed()->mutable_id()->set_node_name("y"); in SumConfig()
Dxla_jit_compiled_cpu_function_test.cc61 tf2xla::Feed* x = config.add_feed(); in SumConfig()
64 tf2xla::Feed* y = config.add_feed(); in SumConfig()
/external/tensorflow/tensorflow/cc/client/
Dclient_session_test.cc107 options.add_feed(a.node()->name()); in TEST()
108 options.add_feed(b.node()->name()); in TEST()
/external/tensorflow/tensorflow/compiler/aot/
Dcodegen_test.cc169 tf2xla::Feed* feed = config.add_feed(); in TEST()
172 feed = config.add_feed(); in TEST()
/external/tensorflow/tensorflow/python/client/
Dsession_ref.cc140 NamedTensorProto* feed = req->add_feed(); in RecordRun()
262 req->add_feed(input); in RecordPRunSetup()
288 auto* feed = req->add_feed(); in RecordPRun()
334 tensor.AsProtoField(req->add_feed()); in RecordRunCallable()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers_test.cc43 request->add_feed("feed_a:0", TensorA()); in BuildRunStepRequest()
44 request->add_feed("feed_b:0", TensorB()); in BuildRunStepRequest()
Dmessage_wrappers.cc72 void InMemoryRunStepRequest::add_feed(const string& name, const Tensor& value) { in add_feed() function in tensorflow::InMemoryRunStepRequest
119 auto feed = proto_version_->add_feed(); in ToProto()
169 void MutableProtoRunStepRequest::add_feed(const string& name, in add_feed() function in tensorflow::MutableProtoRunStepRequest
171 NamedTensorProto* feed = request_.add_feed(); in add_feed()
Dmessage_wrappers.h106 virtual void add_feed(const string& name, const Tensor& value) = 0;
137 void add_feed(const string& name, const Tensor& value) override;
188 void add_feed(const string& name, const Tensor& value) override;
Dmaster_test.cc103 auto f = req.add_feed(); in RunStep()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.cc206 req->add_feed(it.first, it.second); in RunHelper()
308 req.add_feed(feed); in PRunSetup()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_test.cc65 ret.add_feed(feed); in MakeCallableOptions()
334 callable_options.add_feed(y_ + ":0"); in TEST_F()
384 callable_options.add_feed(x_); in TEST_F()
1936 opts.add_feed("x:0"); in TEST()
2009 opts.add_feed("x:0"); in TestFeedAndFetchTensorsInDeviceMemory()
2065 opts.add_feed("x:0"); in TestFeedAndFetchTensorsInDeviceMemoryFailsToMakeCallable()
2203 callable_options.add_feed(input.first); in FeedFetchBenchmarkHelper()
Ddirect_session.cc1423 callable_options.add_feed(input); in GetOrCreateExecutors()
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.cc124 callable_options.add_feed(name); in RunOnce()