Searched refs:mutable_cost_graph (Results 1 – 12 of 12) sorted by relevance
58 cost_graph = run_metadata->mutable_cost_graph(); in PredictCosts()108 metadata.mutable_cost_graph()->Swap(cost_graph); in PredictCosts()
150 cost_graph = run_metadata->mutable_cost_graph(); in PredictCosts()
107 run_graph_response->mutable_cost_graph()->add_node()->set_name("cost_node"); in BuildRunGraphResponse()125 ASSERT_EQ(1, response->mutable_cost_graph()->node_size()); in CheckRunGraphResponse()126 EXPECT_EQ("cost_node", response->mutable_cost_graph()->node(0).name()); in CheckRunGraphResponse()
480 virtual CostGraphDef* mutable_cost_graph() = 0;515 CostGraphDef* mutable_cost_graph() override;548 CostGraphDef* mutable_cost_graph() override;575 CostGraphDef* mutable_cost_graph() override;
620 CostGraphDef* InMemoryRunGraphResponse::mutable_cost_graph() { in mutable_cost_graph() function in tensorflow::InMemoryRunGraphResponse688 CostGraphDef* OwnedProtoRunGraphResponse::mutable_cost_graph() { in mutable_cost_graph() function in tensorflow::OwnedProtoRunGraphResponse689 return response_.mutable_cost_graph(); in mutable_cost_graph()759 CostGraphDef* NonOwnedProtoRunGraphResponse::mutable_cost_graph() { in mutable_cost_graph() function in tensorflow::NonOwnedProtoRunGraphResponse760 return response_->mutable_cost_graph(); in mutable_cost_graph()
416 cost_graph = response->mutable_cost_graph(); in ExecuteAsync()
732 CostGraphDef* cost_graph = run_graph_resp->mutable_cost_graph(); in RunPartitionsHelper()734 resp->mutable_metadata()->mutable_cost_graph()->add_node()->Swap( in RunPartitionsHelper()
161 for (auto node : *init_metadata_.mutable_cost_graph()->mutable_node()) { in Run()200 MergeCosts(metadata->mutable_cost_graph(), init_metadata_.cost_graph(), in Run()
140 metadata1.mutable_cost_graph()->mutable_node(i)->set_compute_cost(0); in TEST_F()144 metadata2.mutable_cost_graph()->mutable_node(i)->set_compute_cost(0); in TEST_F()
233 cost_graph_->Swap(metadata.mutable_cost_graph()); in RealRun()
51 cost_graph->Swap(run_metadata.mutable_cost_graph()); in PredictCosts()
706 CostGraphDef* cost_graph = run_metadata->mutable_cost_graph(); in RunInternal()