| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_def_builder.cc | 38 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()
|
| D | graph_def_builder.h | 81 Options WithControlInput(Node* control_input) const;
127 Options WithControlInputImpl(Node* control_input);
|
| D | node_builder.cc | 138 for (Node* control_input : control_inputs_) { in Finalize()
139 graph->AddControlEdge(control_input, node); in Finalize()
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | subscribe.py | 86 for control_input in op.control_inputs:
87 if control_input not in control_outputs:
88 control_outputs[control_input] = set()
89 control_outputs[control_input].add(op)
|
| D | ops.py | 1855 for control_input in control_inputs:
1856 pywrap_tf_session.TF_AddControlInput(op_desc, control_input._c_op)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 278 def _control_input(self, control_input): argument
279 if control_input is not None and not self._use_ordering_token():
280 return ops.control_dependencies([control_input])
337 control_input=None, argument
356 self._control_input(control_input):
424 control_input = outputs[-1]
426 control_input = None
428 array_ops.concat(flat_tensors, axis=0), control_input,
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_outline_island/ |
| D | case_op.mlir | 3 // CHECK: func @control_input
13 func @control_input(%arg0: tensor<i1>) -> tensor<i32> {
|
| D | while_op.mlir | 3 // CHECK: func @control_input
13 func @control_input(%arg0: tensor<i1>) -> tensor<i32> {
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | single_machine.cc | 416 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/mlir/tensorflow/tests/executor_tpuv1_island_coarsening/ |
| D | while_op.mlir | 6 // CHECK-LABEL: func @control_input
7 func @control_input(%arg0 : tensor<i1>) -> tensor<i32> {
|
| D | executor_tpuv1_island_coarsening.mlir | 5 // CHECK-LABEL: func @control_input
6 func @control_input(%arg0 : tensor<i1>) -> tensor<f32> {
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | encapsulate_util.cc | 293 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/ |
| D | node_def_builder.cc | 253 for (const auto& control_input : control_inputs_) { in Finalize() local
254 node_def->add_input(strings::StrCat("^", control_input)); in Finalize()
|
| D | cost_graph.proto | 73 repeated int32 control_input = 8; field
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils_test.cc | 487 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/compiler/mlir/tensorflow/tests/executor_tpuv1_island_inlining/ |
| D | while_op.mlir | 7 func @control_input(%arg0: tensor<i1>) -> tensor<i32> {
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | loop_optimizer.cc | 107 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/core/grappler/costs/ |
| D | virtual_scheduler.cc | 673 bool control_input = false; in CreateSendRecv() local
678 control_input = true; in CreateSendRecv()
715 if (from->attr().contains(kStreaming) && !control_input) { in CreateSendRecv()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | executor_island_coarsening.mlir | 5 // CHECK-LABEL: func @control_input
7 func @control_input(%arg0 : tensor<i1>) -> tensor<f32> {
|