Searched refs:tensors_to_send (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | rendezvous_util.cc | 25 const std::vector<string>& keys, gtl::ArraySlice<Tensor> tensors_to_send) { in SendTensorsToRendezvous() argument
26 if (keys.size() != tensors_to_send.size()) { in SendTensorsToRendezvous()
29 keys.size(), "; tensors_to_send.size() = ", tensors_to_send.size()); in SendTensorsToRendezvous()
51 rendezvous->Send(parsed, rendez_args, tensors_to_send[i], false)); in SendTensorsToRendezvous()
|
| D | rendezvous_util.h | 36 const std::vector<string>& keys, gtl::ArraySlice<Tensor> tensors_to_send);
|
| D | process_function_library_runtime.h | 59 gtl::ArraySlice<Tensor> tensors_to_send,
|
| D | process_function_library_runtime.cc | 113 gtl::ArraySlice<Tensor> tensors_to_send, DeviceContext* device_context, in SendTensors() argument
117 for (int i = 0; i < tensors_to_send.size(); ++i) { in SendTensors()
124 rendezvous, device_context, alloc_attrs, keys, tensors_to_send)); in SendTensors()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 342 std::vector<Tensor> tensors_to_send; in SendInputs() local
344 tensors_to_send.reserve(in.size()); in SendInputs()
347 tensors_to_send.push_back(p.second); in SendInputs()
350 SendTensorsToRendezvous(rendezvous, nullptr, {}, keys, tensors_to_send); in SendInputs()
437 std::vector<Tensor> tensors_to_send; in ExecuteAsync() local
439 tensors_to_send.reserve(in.size()); in ExecuteAsync()
442 tensors_to_send.push_back(p.second); in ExecuteAsync()
444 s = SendTensorsToRendezvous(rendezvous, nullptr, {}, keys, tensors_to_send); in ExecuteAsync()
|