Searched refs:run_graph_response (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | message_wrappers_test.cc | 102 void BuildRunGraphResponse(MutableRunGraphResponseWrapper* run_graph_response) { in BuildRunGraphResponse() argument
103 run_graph_response->AddRecv("recv_2", TensorA()); in BuildRunGraphResponse()
104 run_graph_response->AddRecv("recv_3", TensorB()); in BuildRunGraphResponse()
105 run_graph_response->mutable_step_stats()->add_dev_stats()->set_device( in BuildRunGraphResponse()
107 run_graph_response->mutable_cost_graph()->add_node()->set_name("cost_node"); in BuildRunGraphResponse()
111 run_graph_response->AddPartitionGraph(graph_def); in BuildRunGraphResponse()
133 void BuildRunStepResponse(MutableRunGraphResponseWrapper* run_graph_response, in BuildRunStepResponse() argument
136 "fetch_x:0", run_graph_response, 0)); in BuildRunStepResponse()
138 "fetch_y:0", run_graph_response, 1)); in BuildRunStepResponse()
140 *run_graph_response->mutable_step_stats(); in BuildRunStepResponse()
[all …]
|
| D | message_wrappers.h | 624 const string& name, MutableRunGraphResponseWrapper* run_graph_response,
659 const string& name, MutableRunGraphResponseWrapper* run_graph_response,
688 const string& name, MutableRunGraphResponseWrapper* run_graph_response,
713 const string& name, MutableRunGraphResponseWrapper* run_graph_response,
|
| D | message_wrappers.cc | 861 const string& name, MutableRunGraphResponseWrapper* run_graph_response, in AddTensorFromRunGraphResponse() argument
865 return run_graph_response->RecvValue(i, response_tensor->mutable_tensor()); in AddTensorFromRunGraphResponse()
913 const string& name, MutableRunGraphResponseWrapper* run_graph_response, in AddTensorFromRunGraphResponse() argument
917 return run_graph_response->RecvValue(i, response_tensor->mutable_tensor()); in AddTensorFromRunGraphResponse()
|