Searched refs:grappler_item_ (Results 1 – 3 of 3) sorted by relevance
712 grappler_item_ = absl::make_unique<GrapplerItem>(); in CreateGrapplerItemWithConv2Ds()713 TF_CHECK_OK(s.ToGraphDef(&grappler_item_->graph)); in CreateGrapplerItemWithConv2Ds()714 grappler_item_->id = "test_conv2d_graph"; in CreateGrapplerItemWithConv2Ds()715 grappler_item_->fetch = {"c0", "c1"}; in CreateGrapplerItemWithConv2Ds()731 grappler_item_ = absl::make_unique<GrapplerItem>(); in CreateGrapplerItemWithConv2DAndVariable()732 TF_CHECK_OK(s.ToGraphDef(&grappler_item_->graph)); in CreateGrapplerItemWithConv2DAndVariable()733 grappler_item_->id = "test_conv2d_var_graph"; in CreateGrapplerItemWithConv2DAndVariable()735 grappler_item_->fetch = {"y"}; in CreateGrapplerItemWithConv2DAndVariable()759 grappler_item_ = absl::make_unique<GrapplerItem>(); in CreateGrapplerItemWithMatmulChain()760 TF_CHECK_OK(s.ToGraphDef(&grappler_item_->graph)); in CreateGrapplerItemWithMatmulChain()[all …]
430 grappler_item_ = item; in Init()431 const auto& graph = grappler_item_->graph; in Init()432 const auto& fetch_nodes = grappler_item_->fetch; in Init()435 for (const auto& f : grappler_item_->feed) { in Init()771 if (grappler_item_->graph.has_library()) { in CreateOpContext()772 op_context.function_library = &grappler_item_->graph.library(); in CreateOpContext()
378 const GrapplerItem* GetGrapplerItem() { return grappler_item_; } in GetGrapplerItem()440 const GrapplerItem* grappler_item_; // Not owned. variable