| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | dependency_optimizer_test.cc | 70 ops::Identity(s.WithOpName("id1").WithControlDependencies(x), add); in TEST_F()
72 s.WithOpName("id2").WithControlDependencies(y).WithControlDependencies(z), in TEST_F()
106 ops::Identity(s.WithOpName("id1").WithControlDependencies(add), x); in TEST_F()
108 ops::Identity(s.WithOpName("id2").WithControlDependencies(add), y); in TEST_F()
160 Output c1 = ops::Const(scope.WithOpName("c1").WithControlDependencies(neg), in TEST_F()
167 ops::Const(scope.WithOpName("c2").WithControlDependencies(ctrl_dep_id), in TEST_F()
206 ops::Identity(s.WithOpName("id1").WithControlDependencies(add), x); in TEST_F()
208 ops::Identity(s.WithOpName("id2").WithControlDependencies(add), y); in TEST_F()
226 auto noop2 = ops::NoOp(s.WithControlDependencies(x)); in TEST_F()
227 Output id = ops::Identity(s.WithControlDependencies({noop1.operation}), x); in TEST_F()
[all …]
|
| D | constant_folding_test.cc | 73 Output c1 = ops::Const(s.WithOpName("c1").WithControlDependencies(add_child), in TEST_F()
593 ops::Const(scope.WithOpName("c").WithControlDependencies(p1), 10, {3}); in TEST_F()
596 ops::Identity(scope.WithOpName("i2").WithControlDependencies(p2), {i1}); in TEST_F()
637 ops::Const(scope.WithOpName("c").WithControlDependencies(p1), 10, {3}); in TEST_F()
640 ops::Identity(scope.WithOpName("i2").WithControlDependencies(p2), {i1}); in TEST_F()
692 ops::Const(scope.WithOpName("c").WithControlDependencies(p1), 10, {3}); in TEST_F()
694 .WithControlDependencies(p2) in TEST_F()
695 .WithControlDependencies(p1), in TEST_F()
1097 ops::Const(scope.WithOpName("const1").WithControlDependencies(x), 2.7f, in TEST_F()
1102 ops::Const(scope.WithOpName("const3").WithControlDependencies(x), 3.14f, in TEST_F()
[all …]
|
| D | model_pruner_test.cc | 136 Output d = ops::AddN(s.WithOpName("d").WithControlDependencies(b), {c}); in TEST_F()
177 scope.WithOpName("id1").WithControlDependencies(v_ctrl), s.output_false); in TEST_F()
180 ops::Identity(scope.WithOpName("id3").WithControlDependencies(id0), id1); in TEST_F()
|
| D | arithmetic_optimizer_test.cc | 101 Output div = ops::Div(s.WithOpName("div").WithControlDependencies( in TEST_F()
167 Output mul = ops::Mul(s.WithControlDependencies(d).WithOpName("mul"), c, c); in TEST_F()
241 s.WithOpName("recip2").WithControlDependencies(squeeze), c); in TEST_F()
302 Output add = ops::Add(s.WithOpName("add").WithControlDependencies(y), x, x); in TEST_F()
855 ops::Identity(s.WithOpName("outputs").WithControlDependencies(transpose2), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow_test.cc | 76 scope.WithOpName("cond").WithControlDependencies(identity_t), 17); in TEST()
84 scope.WithOpName("cond").WithControlDependencies(identity_f), 23); in TEST()
130 scope.WithOpName("cond").WithControlDependencies(identity), 17); in TEST()
153 scope.WithOpName("cond_1").WithControlDependencies(identity), 23); in TEST()
205 scope.WithOpName("while/Less/y").WithControlDependencies(merge.output), in TEST()
216 scope.WithOpName("while/add/y").WithControlDependencies(identity), 1); in TEST()
265 scope.WithOpName("while/Less/y").WithControlDependencies(arg), 10); in TEST()
286 scope.WithOpName("while/add/y").WithControlDependencies(identity), 1); in TEST()
320 scope.WithOpName("while/Less/y").WithControlDependencies(merge.output), in TEST()
329 scope.WithOpName("while/add/y").WithControlDependencies(identity), 1); in TEST()
[all …]
|
| D | xla_compiler_test.cc | 694 scope.WithControlDependencies(std::vector<Operation>{write}), var, in TEST_F()
754 scope.WithControlDependencies(std::vector<Operation>{write}), var, in TEST_F()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition_test.cc | 275 Combine(in_.WithOpName("B2").WithControlDependencies(a1), b1, b1); in TEST_F()
283 auto c = Const(scope_a_.WithOpName("A1/_0").WithControlDependencies(a1), {}); in TEST_F()
291 Combine(scope_b_.WithOpName("B2").WithControlDependencies(id), b1, b1); in TEST_F()
321 Combine(in_.WithOpName("B2").WithControlDependencies(a1), b1, b1); in TEST_F()
322 FloatInput(in_.WithOpName("B3").WithControlDependencies(a1)); in TEST_F()
330 auto c = Const(scope_a_.WithOpName("A1/_0").WithControlDependencies(a1), {}); in TEST_F()
338 Combine(scope_b_.WithOpName("B2").WithControlDependencies(id), b1, b1); in TEST_F()
339 FloatInput(scope_b_.WithOpName("B3").WithControlDependencies(id)); in TEST_F()
347 FloatInput(in_.WithOpName("B3").WithControlDependencies(a1)); in TEST_F()
355 auto c = Const(scope_a_.WithOpName("A1/_0").WithControlDependencies(a1), {}); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_test.cc | 659 ops::NoOp(s.WithOpName("Func/_0").WithControlDependencies({c})); in TEST_F()
661 s.WithOpName("Func/_1").WithControlDependencies({func0}), a); in TEST_F()
683 ops::NoOp(s.WithOpName("Func/_0").WithControlDependencies({c})); in TEST_F()
685 s.WithOpName("Func/_1").WithControlDependencies({func0}), a); in TEST_F()
688 ops::NoOp(s.WithOpName("Func/_3").WithControlDependencies({func0})); in TEST_F()
690 s.WithOpName("Func/_4").WithControlDependencies({func3}), func1); in TEST_F()
692 s.WithOpName("b/x2/two").WithControlDependencies({func3}), 2LL); in TEST_F()
698 ops::NoOp(s.WithOpName("Func/_6").WithControlDependencies({func0})); in TEST_F()
700 s.WithOpName("Func/_7").WithControlDependencies({func6}), func5); in TEST_F()
702 s.WithOpName("b/y/two").WithControlDependencies({func6}), 2LL); in TEST_F()
[all …]
|
| D | constant_folding_test.cc | 400 auto c1 = ops::Const<int>(s.WithControlDependencies(recv1), 2); in TEST_F()
403 auto c2 = ops::Const<int>(s.WithControlDependencies(recv2), 3); in TEST_F()
404 auto add = ops::Add(s.WithControlDependencies(c2), c0, c1); in TEST_F()
444 auto c = ops::Const<int>(s.WithControlDependencies(recv2), 3); in TEST_F()
445 auto add0 = ops::Add(s.WithControlDependencies(c), rank, size); in TEST_F()
585 ops::Const<int>(s.WithOpName("c0").WithControlDependencies(recv0), 1); in TEST_F()
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | scope.h | 122 Scope WithControlDependencies(
126 Scope WithControlDependencies(const Output& control_dep) const;
|
| D | scope_test.cc | 132 Scope c = root.WithControlDependencies({c1, c2}); in TEST()
134 Scope c_c = c.WithControlDependencies({Operation()}); in TEST()
|
| D | cc_ops_test.cc | 130 Scope with_control_deps = root.WithControlDependencies(assign); in TEST()
135 IsVariableInitialized(no_control_deps.WithControlDependencies(sub), v); in TEST()
|
| D | scope.cc | 362 Scope Scope::WithControlDependencies( in WithControlDependencies() function in tensorflow::Scope
370 Scope Scope::WithControlDependencies(const Output& control_dep) const { in WithControlDependencies() function in tensorflow::Scope
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | graph_to_functiondef_test.cc | 89 auto b = ops::Neg(root.WithOpName("b").WithControlDependencies(a), a); in TEST()
90 auto c = ops::_Retval(root.WithOpName("c").WithControlDependencies(b), b, 0); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 203 ops::Neg(s.WithControlDependencies(neg0).WithOpName("neg1"), c0); in TEST_F()
207 auto noop = ops::NoOp(s.WithControlDependencies(neg0).WithOpName("noop")); in TEST_F()
|
| D | graph_view_test.cc | 84 Output d = ops::AddN(s.WithOpName("d").WithControlDependencies(a), {b, c}); in TEST_F()
|
| /external/tensorflow/tensorflow/cc/ops/ |
| D | while_loop.cc | 102 scope.NewSubScope("cond").WithControlDependencies(inputs[0]); in CreateCond()
129 scope.NewSubScope("body").WithControlDependencies(inputs[0]); in CreateBody()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 160 auto b = ops::RandomUniform(s.WithOpName("b").WithControlDependencies(a), in CreateGrapplerItemWithMatmulChain()
162 auto c = ops::RandomUniform(s.WithOpName("c").WithControlDependencies(b), in CreateGrapplerItemWithMatmulChain()
164 auto d = ops::RandomUniform(s.WithOpName("d").WithControlDependencies(c), in CreateGrapplerItemWithMatmulChain()
166 auto e = ops::RandomUniform(s.WithOpName("e").WithControlDependencies(d), in CreateGrapplerItemWithMatmulChain()
169 auto ab = ops::MatMul(s.WithOpName("ab").WithControlDependencies(e), a, b); in CreateGrapplerItemWithMatmulChain()
218 ops::NoOp(s.WithControlDependencies(input_tensors).WithOpName("out")); in CreateGrapplerItemWithControlDependency()
259 auto z4 = ops::NoOp(s.WithControlDependencies(batch_var).WithOpName("z4")); in CreateGrapplerItemWithBatchNorm()
823 .WithControlDependencies(y) in CreateGrapplerItemWithInterDeviceTransfers()
|
| D | graph_memory_test.cc | 202 s.WithOpName("init").WithDevice("/CPU:0").WithControlDependencies( in TEST_F()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | scope.go | 111 func (s *Scope) WithControlDependencies(ops ...*tf.Operation) *Scope { func
|
| D | scope_test.go | 83 s = s.WithControlDependencies(readDeps...)
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| D | math_grad.cc | 204 Scope grad_scope = scope.WithControlDependencies(grad); in TanhGrad()
260 Scope grad_scope = scope.WithControlDependencies(grad); in SigmoidGrad()
743 Scope grad_scope = scope.WithControlDependencies(grad); in ErfGrad()
758 Scope grad_scope = scope.WithControlDependencies(grad); in LgammaGrad()
|