Home
last modified time | relevance | path

Searched refs:ControlInput (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/data/
Dhash_utils_test.cc875 .ControlInput(n2->name()) in TEST_F()
882 .ControlInput(n3->name()) in TEST_F()
920 .ControlInput(n2->name()) in TEST_F()
921 .ControlInput(n3->name()) in TEST_F()
928 .ControlInput(n3->name()) in TEST_F()
929 .ControlInput(n2->name()) in TEST_F()
985 ndb.ControlInput(absl::StrCat("graph_1/node_", j)); in TEST_F()
1037 .ControlInput("graph_1/node_2") in TEST_F()
1054 .ControlInput("graph_1/node_2") in TEST_F()
1061 .ControlInput("graph_1/node_1") in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/graph/
Dnode_builder.cc85 NodeBuilder& NodeBuilder::ControlInput(Node* src_node) { in ControlInput() function in tensorflow::NodeBuilder
87 def_builder_.ControlInput(src_node->name()); in ControlInput()
95 def_builder_.ControlInput(src_node->name()); in ControlInputs()
Dnode_builder.h98 NodeBuilder& ControlInput(Node* src_node);
/external/tensorflow/tensorflow/core/common_runtime/
Dparallel_concat_optimizer.cc98 identity_def.ControlInput(s); in Run()
Dlower_while_op.cc433 .ControlInput(lowered_while_executed_) in CreateExitNodes()
451 .ControlInput(body_call_node_) in CreateNextIterationNodes()
Dlower_case_op.cc264 .ControlInput(branch_executed_node_) in BuildLoweredCaseOutput()
Dlower_while_op_test.cc304 .ControlInput(control_in) in TEST()
321 .ControlInput(while_node) in TEST()
Dlower_if_op.cc294 .ControlInput(branch_executed_node_) in BuildLoweredIfOutput()
Dinline_function_utils.cc826 .ControlInput(output_control_node) in InlineFunctionBody()
833 .ControlInput(output_control_node) in InlineFunctionBody()
Dlower_if_op_test.cc202 .ControlInput(init.operation.node()) in TEST()
Dlower_case_op_test.cc239 .ControlInput(init.operation.node()) in TEST()
Ddirect_session_test.cc1425 builder = builder.ControlInput(control_dep); in TEST()
/external/tensorflow/tensorflow/core/framework/
Dnode_def_builder_test.cc141 Builder().ControlInput("x").Input(FakeInput()).ControlInput("y"), in TEST_F()
196 ExpectInvalid(Builder().Input(FakeInput()).ControlInput("z:2"), in TEST_F()
211 .ControlInput("y") in TEST_F()
Dnode_def_builder.h90 NodeDefBuilder& ControlInput(StringPiece src_node);
Dnode_def_builder.cc204 NodeDefBuilder& NodeDefBuilder::ControlInput(StringPiece src_node) { in ControlInput() function in tensorflow::NodeDefBuilder
/external/tensorflow/tensorflow/core/nccl/
Dnccl_rewrite.cc174 send_builder.ControlInput(edge->src()); in ReplaceBroadcast()
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils_test.cc161 .ControlInput("constant") in TEST()
/external/tensorflow/tensorflow/compiler/jit/
Dencapsulate_subgraphs_pass_test.cc684 nb.Input(a).ControlInput(control); in TEST()
688 nb2.Input(b).Input(call1).ControlInput(control); in TEST()
2086 node_builder1.Input(a).Input(b).ControlInput(s1); in TEST()
2231 node_builder1.Input(a).Input(b).ControlInput(s1); in TEST()
2404 node_builder1.Input(a).Input(b).ControlInput(s1); in TEST()
2512 node_builder1.Input(a).Input(b).ControlInput(s); in TEST()
Dxla_compilation_cache.cc237 .ControlInput(graph->source_node()) in CreateGraph()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_while.cc455 builder.ControlInput(in_edge->src()->name()); in FunctionalizeLoop()
Dfunctionalize_cond.cc784 builder.ControlInput(in_edge->src()->name()); in BuildIfNode()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc478 node_builder.ControlInput(c); in CreateTRTNode()
/external/tensorflow/tensorflow/c/
Dc_api.cc808 desc->node_builder.ControlInput(&input->node); in TF_AddControlInput()