Home
last modified time | relevance | path

Searched refs:control_input (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/graph/
Dgraph_def_builder.cc38 Node* control_input) const { in WithControlInput()
39 return Options(*this).WithControlInputImpl(control_input); in WithControlInput()
56 Node* control_input) { in WithControlInputImpl() argument
57 control_inputs_.push_back(control_input); in WithControlInputImpl()
Dgraph_def_builder.h81 Options WithControlInput(Node* control_input) const;
127 Options WithControlInputImpl(Node* control_input);
Dnode_builder.cc138 for (Node* control_input : control_inputs_) { in Finalize()
139 graph->AddControlEdge(control_input, node); in Finalize()
/external/v8/src/compiler/
Dschedule.cc61 void BasicBlock::set_control_input(Node* control_input) { in set_control_input() argument
62 if (!nodes_.empty() && control_input == nodes_.back()) { in set_control_input()
65 control_input_ = control_input; in set_control_input()
312 if (block->control_input() != nullptr) { in InsertBranch()
313 SetControlInput(end, block->control_input()); in InsertBranch()
328 if (block->control_input() != nullptr) { in InsertSwitch()
329 SetControlInput(end, block->control_input()); in InsertSwitch()
495 if (block->control_input() != nullptr) { in operator <<()
496 os << *block->control_input(); in operator <<()
Dbranch-elimination.cc127 Node* control_input = NodeProperties::GetControlInput(node, 0); in ReduceBranch() local
128 ControlPathConditions from_input = node_conditions_.Get(control_input); in ReduceBranch()
137 Replace(use, condition_value ? control_input : dead()); in ReduceBranch()
140 Replace(use, condition_value ? dead() : control_input); in ReduceBranch()
Draw-machine-assembler.cc154 Node* control_input = successor->control_input(); in OptimizeControlFlow() local
155 block->set_control_input(control_input); in OptimizeControlFlow()
156 if (control_input) { in OptimizeControlFlow()
157 schedule->SetBlockForNode(block, control_input); in OptimizeControlFlow()
175 Node* branch = block->control_input(); in OptimizeControlFlow()
248 graph(), common(), block->PredecessorAt(i)->control_input()); in MakeReschedulable()
325 if (Node* block_terminator = block->control_input()) { in MakeReschedulable()
Dstore-store-elimination.cc342 Node* control_input = NodeProperties::GetControlInput(node, i); in Visit() local
343 if (!HasBeenVisited(control_input)) { in Visit()
344 MarkForRevisit(control_input); in Visit()
Dgraph-visualizer.cc685 if (current->control_input() != nullptr) { in PrintSchedule()
686 PrintNode(current->control_input()); in PrintSchedule()
694 if (FLAG_trace_turbo_types && current->control_input() != nullptr) { in PrintSchedule()
696 PrintType(current->control_input()); in PrintSchedule()
944 if (current->control_input() != nullptr) { in PrintScheduledGraph()
945 PrintScheduledNode(os, indent, current->control_input()); in PrintScheduledGraph()
963 DCHECK_NULL(current->control_input()); in PrintScheduledGraph()
Dschedule.h129 Node* control_input() const { return control_input_; } in NON_EXPORTED_BASE()
130 void set_control_input(Node* control_input); in NON_EXPORTED_BASE()
Dgraph-assembler.cc173 original_control_input_ = original_block()->control_input(); in CopyForChange()
191 schedule_->SetBlockForNode(nullptr, original_block()->control_input()); in CopyForChange()
921 Node* throw_node = block->control_input(); in FinalizeCurrentBlock()
Dsimd-scalar-lowering.cc562 Node* control_input = node->InputAt(3); in LowerLoadOp() local
565 control_input); in LowerLoadOp()
643 Node* control_input = node->InputAt(3); in LowerLoadTransformOp() local
661 control_input); in LowerLoadTransformOp()
681 graph()->NewNode(load_op, base, rep, effect_input, control_input); in LowerLoadTransformOp()
757 Node* control_input = node->InputAt(4); in LowerStoreOp() local
761 effect_input, control_input); in LowerStoreOp()
Dmachine-graph-verifier.cc100 i < block->NodeCount() ? block->NodeAt(i) : block->control_input(); in Run()
356 i < block->NodeCount() ? block->NodeAt(i) : block->control_input(); in Run()
Dverifier.cc1894 if (node == block->control_input()) return true; in HasDominatingDef()
2098 Node* control = block->control_input(); in Run()
Draw-machine-assembler.h1046 void MarkControlDeferred(Node* control_input);
/external/tensorflow/tensorflow/python/framework/
Dsubscribe.py85 for control_input in op.control_inputs:
86 if control_input not in control_outputs:
87 control_outputs[control_input] = set()
88 control_outputs[control_input].add(op)
Dops.py1651 for control_input in control_inputs:
1652 pywrap_tf_session.TF_AddControlInput(op_desc, control_input._c_op)
/external/tensorflow/tensorflow/core/grappler/clusters/
Dsingle_machine.cc416 for (auto& control_input : *new_node->mutable_control_input()) { in MergeCosts()
417 control_input += queue_costs_id_offset; in MergeCosts()
436 for (auto& control_input : *new_node->mutable_control_input()) { in MergeCosts()
437 control_input += init_costs_id_offset; in MergeCosts()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_util.cc293 for (const string& control_input : control_deps) { in PostprocessControlEdgesBetweenOutsideCompilations() local
294 auto iter = node_name_index.find(control_input); in PostprocessControlEdgesBetweenOutsideCompilations()
297 control_input); in PostprocessControlEdgesBetweenOutsideCompilations()
/external/tensorflow/tensorflow/core/framework/
Dnode_def_builder.cc253 for (const auto& control_input : control_inputs_) { in Finalize() local
254 node_def->add_input(strings::StrCat("^", control_input)); in Finalize()
Dcost_graph.proto71 repeated int32 control_input = 8; field
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_utils.cc231 for (const NodeDef* control_input : node_rep->node_control_inputs) { in ComputeNodeHash() local
233 EdgeRep edge(node, control_input); in ComputeNodeHash()
237 TF_RETURN_IF_ERROR(ComputeNonInputNodeHash(*control_input, &node_hash)); in ComputeNodeHash()
239 TF_RETURN_IF_ERROR(ComputeNodeHash(control_input, &node_hash)); in ComputeNodeHash()
/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dvectorization_utils_test.cc487 string control_input = strings::StrCat("^", print_node.name()); in TEST() local
488 EXPECT_TRUE(cast_node.input(0) == control_input || in TEST()
489 cast_node.input(1) == control_input); in TEST()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dloop_optimizer.cc107 for (auto& control_input : enter_control_inputs) { in HandleInvariantEnter() local
108 consumer->add_input(control_input); in HandleInvariantEnter()
109 node_map_->AddOutput(NodeName(control_input), consumer->name()); in HandleInvariantEnter()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/
Dexecutor_island_coarsening.mlir5 // CHECK-LABEL: func @control_input
7 func @control_input(%arg0 : tensor<i1>) -> tensor<f32> {
/external/v8/src/compiler/backend/
Dinstruction-selector.cc436 cont->true_block()->PredecessorAt(0)->control_input()) in GetEffectLevel()
1145 if (block->control_input() != nullptr) { in VisitBlock()
1146 SetEffectLevel(block->control_input(), effect_level); in VisitBlock()
1167 if (!FinishEmittedInstructions(block->control_input(), current_block_end)) in VisitBlock()
1219 Node* input = block->control_input(); in VisitControl()

12