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()
664 GraphDef* InMemoryRunGraphResponse::mutable_partition_graph(size_t i) { in mutable_partition_graph() function in tensorflow::InMemoryRunGraphResponse730 GraphDef* OwnedProtoRunGraphResponse::mutable_partition_graph(size_t i) { in mutable_partition_graph() function in tensorflow::OwnedProtoRunGraphResponse731 return response_.mutable_partition_graph(i); in mutable_partition_graph()736 GraphDef* graph_def = response_.mutable_partition_graph()->Add(); in AddPartitionGraph()803 GraphDef* NonOwnedProtoRunGraphResponse::mutable_partition_graph(size_t i) { in mutable_partition_graph() function in tensorflow::NonOwnedProtoRunGraphResponse804 return response_->mutable_partition_graph(i); in mutable_partition_graph()
494 virtual GraphDef* mutable_partition_graph(size_t i) = 0;529 GraphDef* mutable_partition_graph(size_t i) override;562 GraphDef* mutable_partition_graph(size_t i) override;589 GraphDef* mutable_partition_graph(size_t i) override;
800 run_graph_resp->mutable_partition_graph(i)); in RunPartitionsHelper()