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 | 94 gtl::ArraySlice<Tensor> tensors_to_send,
|
| D | process_function_library_runtime.cc | 123 gtl::ArraySlice<Tensor> tensors_to_send, DeviceContext* device_context, in SendTensors() argument
127 for (int i = 0; i < tensors_to_send.size(); ++i) { in SendTensors()
134 rendezvous, device_context, alloc_attrs, keys, tensors_to_send)); in SendTensors()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.cc | 359 std::vector<Tensor> tensors_to_send; in SendInputs() local
361 tensors_to_send.reserve(in.size()); in SendInputs()
365 tensors_to_send.push_back(p.second); in SendInputs()
370 SendTensorsToRendezvous(rendezvous, nullptr, {}, keys, tensors_to_send); in SendInputs()
474 std::vector<Tensor> tensors_to_send; in ExecuteAsync() local
476 tensors_to_send.reserve(in.size()); in ExecuteAsync()
479 tensors_to_send.push_back(p.second); in ExecuteAsync()
482 s = SendTensorsToRendezvous(rendezvous, nullptr, {}, keys, tensors_to_send); in ExecuteAsync()
|