Searched refs:ctrl_dep (Results 1 – 5 of 5) sorted by relevance
94 const string ctrl_dep = in MaybeAddControlInput() local96 node->add_input(ctrl_dep); in MaybeAddControlInput()447 string ctrl_dep = in MaterializeShapes() local449 node_map_->UpdateInput(node->name(), node->input(0), ctrl_dep); in MaterializeShapes()450 node->set_input(0, ctrl_dep); in MaterializeShapes()538 string ctrl_dep = AddControlDependency(shape_n_node->name(), in MaterializeShapes() local540 *added_node->add_input() = ctrl_dep; in MaterializeShapes()541 node_map_->AddOutput(NodeName(ctrl_dep), added_node->name()); in MaterializeShapes()702 string ctrl_dep = in MaterializeBroadcastGradientArgs() local704 *out[j]->add_input() = ctrl_dep; in MaterializeBroadcastGradientArgs()[all …]
180 const string ctrl_dep = ConstantFolding::AddControlDependency( in HandleConst() local182 const_node->add_input(ctrl_dep); in HandleConst()183 node_map_->AddOutput(NodeName(ctrl_dep), const_node->name()); in HandleConst()539 const string ctrl_dep = ConstantFolding::AddControlDependency( in RemoveStackOps() local541 push_node->set_input(1, ctrl_dep); in RemoveStackOps()
681 for (const string& ctrl_dep : pre_ctrl_deps) { in SchedulingPass() local682 *tmp_var->add_input() = ctrl_dep; in SchedulingPass()733 for (const string& ctrl_dep : post_ctrl_deps) { in SchedulingPass() local734 *node->add_input() = ctrl_dep; in SchedulingPass()
124 const string ctrl_dep = in MaybeAddControlInput() local126 node->add_input(ctrl_dep); in MaybeAddControlInput()762 for (const string& ctrl_dep : ctrl_deps) { in TrySimplify() local763 *new_add_node->add_input() = ctrl_dep; in TrySimplify()764 ctx().node_map->AddOutput(NodeName(ctrl_dep), new_add_node->name()); in TrySimplify()
314 auto ctrl_dep = in TEST_F() local320 NodeWith(Inputs(Out(NodeWith(Op("IdentityN"), CtrlDeps(ctrl_dep)))))); in TEST_F()