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.cc51 tf2xla::Feed* feed = config.add_feed(); in TEST()
55 feed = config.add_feed(); in TEST()
75 tf2xla::Feed* feed = config.add_feed(); in TEST()
82 config.add_feed(); in TEST()
88 tf2xla::Feed* feed = config.add_feed(); in TEST()
96 tf2xla::Feed* feed = config.add_feed(); in TEST()
104 tf2xla::Feed* feed = config.add_feed(); in TEST()
114 tf2xla::Feed* feed = config.add_feed(); in TEST()
117 feed = config.add_feed(); in TEST()
125 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()
109 config.add_feed()->mutable_id()->set_node_name("unused"); in TEST()
Dxla_jit_compiled_cpu_function_test.cc74 tf2xla::Feed* x = config.add_feed(); in SumConfig()
77 tf2xla::Feed* y = config.add_feed(); in SumConfig()
/external/tensorflow/tensorflow/cc/client/
Dclient_session_test.cc193 options.add_feed(a.node()->name()); in TEST()
194 options.add_feed(b.node()->name()); in TEST()
229 options.add_feed(a.node()->name()); in TEST()
230 options.add_feed(b.node()->name()); in TEST()
/external/tensorflow/tensorflow/compiler/aot/
Dcodegen_test.cc193 tf2xla::Feed* feed = config.add_feed(); in TEST()
196 feed = config.add_feed(); in TEST()
/external/tensorflow/tensorflow/python/client/
Dsession_ref.cc141 NamedTensorProto* feed = req->add_feed(); in RecordRun()
263 req->add_feed(input); in RecordPRunSetup()
289 auto* feed = req->add_feed(); in RecordPRun()
335 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.cc70 void InMemoryRunStepRequest::add_feed(const string& name, const Tensor& value) { in add_feed() function in tensorflow::InMemoryRunStepRequest
117 auto feed = proto_version_->add_feed(); in ToProto()
167 void MutableProtoRunStepRequest::add_feed(const string& name, in add_feed() function in tensorflow::MutableProtoRunStepRequest
169 NamedTensorProto* feed = request_.add_feed(); in add_feed()
Dmessage_wrappers.h109 virtual void add_feed(const string& name, const Tensor& value) = 0;
140 void add_feed(const string& name, const Tensor& value) override;
191 void add_feed(const string& name, const Tensor& value) override;
Dmaster_test.cc104 auto f = req.add_feed(); in RunStep()
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session.cc223 req->add_feed(it.first, it.second); in RunHelper()
325 req.add_feed(feed); in PRunSetup()
/external/tensorflow/tensorflow/core/common_runtime/
Ddirect_session_test.cc71 ret.add_feed(feed); in MakeCallableOptions()
481 callable_options.add_feed(y_ + ":0"); in TEST_F()
531 callable_options.add_feed(x_); in TEST_F()
2365 opts.add_feed("x:0"); in TEST()
2438 opts.add_feed("x:0"); in TestFeedAndFetchTensorsInDeviceMemory()
2494 opts.add_feed("x:0"); in TestFeedAndFetchTensorsInDeviceMemoryFailsToMakeCallable()
2636 callable_options.add_feed(input.first); in FeedFetchBenchmarkHelper()
Ddirect_session.cc1539 callable_options.add_feed(input); in GetOrCreateExecutors()
/external/tensorflow/tensorflow/cc/saved_model/
Dloader.cc163 callable_options.add_feed(name); in RunOnce()