Searched refs:ctrl_dep (Results 1 – 4 of 4) sorted by relevance
111 const string ctrl_dep = in MaybeAddControlInput() local113 node->add_input(ctrl_dep); in MaybeAddControlInput()396 string ctrl_dep = in MaterializeShapes() local398 node->set_input(0, ctrl_dep); in MaterializeShapes()399 node_map_->AddOutput(NodeName(ctrl_dep), node->name()); in MaterializeShapes()485 string ctrl_dep = AddControlDependency(shape_n_node->name(), in MaterializeShapes() local487 *added_node->add_input() = ctrl_dep; in MaterializeShapes()488 node_map_->AddOutput(NodeName(ctrl_dep), added_node->name()); in MaterializeShapes()648 string ctrl_dep = in MaterializeBroadcastGradientArgs() local650 *out[j]->add_input() = ctrl_dep; in MaterializeBroadcastGradientArgs()[all …]
182 const string ctrl_dep = ConstantFolding::AddControlDependency( in HandleConst() local184 const_node->add_input(ctrl_dep); in HandleConst()185 node_map_->AddOutput(NodeName(ctrl_dep), const_node->name()); in HandleConst()540 const string ctrl_dep = ConstantFolding::AddControlDependency( in RemoveStackOps() local542 push_node->set_input(1, ctrl_dep); in RemoveStackOps()
650 for (const string& ctrl_dep : pre_ctrl_deps) { in SchedulingPass() local651 *tmp_var->add_input() = ctrl_dep; in SchedulingPass()702 for (const string& ctrl_dep : post_ctrl_deps) { in SchedulingPass() local703 *node->add_input() = ctrl_dep; in SchedulingPass()
122 const string ctrl_dep = in MaybeAddControlInput() local124 node->add_input(ctrl_dep); in MaybeAddControlInput()736 for (const string& ctrl_dep : ctrl_deps) { in TrySimplify() local737 *new_add_node->add_input() = ctrl_dep; in TrySimplify()738 ctx().node_map->AddOutput(NodeName(ctrl_dep), new_add_node->name()); in TrySimplify()