Searched refs:mutable_partition_graph (Results 1 – 4 of 4) sorted by relevance
128 EXPECT_EQ(1234, response->mutable_partition_graph(0)->versions().producer()); in CheckRunGraphResponse()130 response->mutable_partition_graph(0)->versions().min_consumer()); in CheckRunGraphResponse()145 run_graph_response->mutable_partition_graph(i)); in BuildRunStepResponse()
645 GraphDef* InMemoryRunGraphResponse::mutable_partition_graph(size_t i) { in mutable_partition_graph() function in tensorflow::InMemoryRunGraphResponse711 GraphDef* OwnedProtoRunGraphResponse::mutable_partition_graph(size_t i) { in mutable_partition_graph() function in tensorflow::OwnedProtoRunGraphResponse712 return response_.mutable_partition_graph(i); in mutable_partition_graph()717 GraphDef* graph_def = response_.mutable_partition_graph()->Add(); in AddPartitionGraph()784 GraphDef* NonOwnedProtoRunGraphResponse::mutable_partition_graph(size_t i) { in mutable_partition_graph() function in tensorflow::NonOwnedProtoRunGraphResponse785 return response_->mutable_partition_graph(i); in mutable_partition_graph()
482 virtual GraphDef* mutable_partition_graph(size_t i) = 0;517 GraphDef* mutable_partition_graph(size_t i) override;550 GraphDef* mutable_partition_graph(size_t i) override;577 GraphDef* mutable_partition_graph(size_t i) override;
743 run_graph_resp->mutable_partition_graph(i)); in RunPartitionsHelper()