Searched refs:grappler_item_ (Results 1 – 3 of 3) sorted by relevance
133 grappler_item_.reset(new GrapplerItem); in CreateGrapplerItemWithConv2Ds()134 grappler_item_->id = "test_conv2d_graph"; in CreateGrapplerItemWithConv2Ds()135 grappler_item_->graph = def; in CreateGrapplerItemWithConv2Ds()136 grappler_item_->fetch = {"c0", "c1"}; in CreateGrapplerItemWithConv2Ds()154 grappler_item_.reset(new GrapplerItem); in CreateGrapplerItemWithConv2DAndVariable()155 grappler_item_->id = "test_conv2d_var_graph"; in CreateGrapplerItemWithConv2DAndVariable()156 grappler_item_->graph = def; in CreateGrapplerItemWithConv2DAndVariable()157 grappler_item_->fetch = {"y"}; in CreateGrapplerItemWithConv2DAndVariable()184 grappler_item_.reset(new GrapplerItem); in CreateGrapplerItemWithMatmulChain()185 grappler_item_->id = "test_matmul_sequence_graph"; in CreateGrapplerItemWithMatmulChain()[all …]
275 grappler_item_ = item; in Init()307 const auto& graph = grappler_item_->graph; in Init()308 const auto& fetch_nodes = grappler_item_->fetch; in Init()310 for (const auto& f : grappler_item_->feed) { in Init()646 if (grappler_item_->graph.has_library()) { in GetCurrNode()647 op_context.function_library = &grappler_item_->graph.library(); in GetCurrNode()
353 const GrapplerItem* grappler_item_; // Not owned. variable