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()
178 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()
639 CostGraphDef* InMemoryRunGraphResponse::mutable_cost_graph() { in mutable_cost_graph() function in tensorflow::InMemoryRunGraphResponse707 CostGraphDef* OwnedProtoRunGraphResponse::mutable_cost_graph() { in mutable_cost_graph() function in tensorflow::OwnedProtoRunGraphResponse708 return response_.mutable_cost_graph(); in mutable_cost_graph()778 CostGraphDef* NonOwnedProtoRunGraphResponse::mutable_cost_graph() { in mutable_cost_graph() function in tensorflow::NonOwnedProtoRunGraphResponse779 return response_->mutable_cost_graph(); in mutable_cost_graph()
492 virtual CostGraphDef* mutable_cost_graph() = 0;527 CostGraphDef* mutable_cost_graph() override;560 CostGraphDef* mutable_cost_graph() override;587 CostGraphDef* mutable_cost_graph() override;
448 cost_graph = response->mutable_cost_graph(); in ExecuteAsync()
789 CostGraphDef* cost_graph = run_graph_resp->mutable_cost_graph(); in RunPartitionsHelper()791 resp->mutable_metadata()->mutable_cost_graph()->add_node()->Swap( in RunPartitionsHelper()
160 for (auto node : *init_metadata_.mutable_cost_graph()->mutable_node()) { in Run()194 MergeCosts(metadata->mutable_cost_graph(), init_metadata_.cost_graph(), in Run()
153 metadata1.mutable_cost_graph()->mutable_node(i)->set_compute_cost(0); in TEST_F()157 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()
791 CostGraphDef* cost_graph = run_metadata->mutable_cost_graph(); in RunInternal()