| /external/tensorflow/tensorflow/core/nccl/ |
| nccl_rewrite.cc | 45 std::vector<Node*> control_inputs; local
48 control_inputs.push_back(edge->src());
57 make_builder("_NcclReduceRecv", "Recv").ControlInputs(control_inputs);
74 .ControlInputs(control_inputs);
|
| /external/tensorflow/tensorflow/cc/framework/ |
| scope.cc | 327 std::vector<Node*> control_inputs; local
329 control_inputs.push_back(op.node());
331 builder->ControlInputs(control_inputs);
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| encapsulate_xla_computations_pass.cc | 276 absl::flat_hash_set<Node*> control_inputs; local
279 AddControlInputs(*launch, &control_inputs);
343 control_inputs.erase(node);
357 for (Node* n : control_inputs) {
|
| encapsulate_subgraphs_pass.cc | 379 std::unordered_set<const Node*> control_inputs; member in struct:tensorflow::__anon44060::Encapsulator::Subgraph::OutsideCompilationSubgraph
[all...] |
| /external/tensorflow/tensorflow/core/common_runtime/ |
| executor_test.cc | 433 std::vector<Node*> control_inputs; local
435 control_inputs.push_back(ready_nodes.back());
438 Node* n = test::graph::NoOp(g, control_inputs);
|
| lower_while_op.cc | 209 std::vector<Node*> control_inputs; local
212 control_inputs.push_back(e->src());
215 if (!control_inputs.empty()) {
219 .ControlInputs(control_inputs)
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| single_threaded_executor_test.cc | 264 std::vector<Node*> control_inputs; local
266 control_inputs.push_back(ready_nodes.back());
269 Node* n = test::graph::NoOp(g, control_inputs);
|
| /external/v8/src/compiler/ |
| common-operator-reducer.cc | 344 Node::Inputs control_inputs = control->inputs(); local
346 DCHECK_NE(0, control_inputs.count());
347 DCHECK_EQ(control_inputs.count(), value_inputs.count() - 1);
351 for (int i = 0; i < control_inputs.count(); ++i) {
357 effect, control_inputs[i]);
366 DCHECK_EQ(control_inputs.count(), effect_inputs.count() - 1);
367 for (int i = 0; i < control_inputs.count(); ++i) {
373 effect_inputs[i], control_inputs[i]);
|
| /external/tensorflow/tensorflow/core/graph/ |
| graph_constructor.cc | 717 std::set<TensorId> control_inputs; local
729 if (control_inputs.count(new_input) > 0) {
733 control_inputs.insert(new_input);
[all...] |
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| pfor.py | 185 self._next_iter_control_inputs.append(next_iter.control_inputs)
217 def control_inputs(self): member in class:WhileOp
219 control_inputs = []
221 control_inputs.extend(x.op.control_inputs)
222 return control_inputs
325 control_inputs = [
326 parent_pfor._convert_helper(x).t for x in enter.op.control_inputs
328 if control_inputs:
329 with ops.control_dependencies(control_inputs)
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| function.cc | 361 for (int index : node_info.control_inputs) {
437 nodes_[node_index].control_inputs.push_back(dep_index);
451 std::vector<int> control_inputs; member in struct:tensorflow::__anon44961::FunctionInstantiationHelper::NodeInfo
[all...] |
| /external/tensorflow/tensorflow/c/ |
| c_api_test.cc | 782 TF_Operation* control_inputs[100]; local
784 scalar3, control_inputs, TF_OperationNumControlInputs(scalar3));
786 EXPECT_EQ(feed, control_inputs[0]);
787 EXPECT_EQ(feed2, control_inputs[1]);
790 feed3, control_inputs, TF_OperationNumControlInputs(feed3));
792 EXPECT_EQ(feed, control_inputs[0]);
793 EXPECT_EQ(feed2, control_inputs[1]);
817 scalar4, control_inputs, TF_OperationNumControlInputs(scalar4));
819 EXPECT_EQ(feed, control_inputs[0]);
820 EXPECT_EQ(feed4, control_inputs[1])
[all...] |
| /external/tensorflow/tensorflow/python/framework/ |
| ops.py | 2397 def control_inputs(self): member in class:Operation
4541 def control_inputs(self): member in class:Graph._ControlDependenciesController
[all...] |
