| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | dependency_optimizer_test.cc | 74 ops::Identity(s.WithOpName("id1").WithControlDependencies(x), add); in TEST_F()
76 s.WithOpName("id2").WithControlDependencies(y).WithControlDependencies(z), in TEST_F()
110 ops::Identity(s.WithOpName("id1").WithControlDependencies(add), x); in TEST_F()
112 ops::Identity(s.WithOpName("id2").WithControlDependencies(add), y); in TEST_F()
156 ops::Identity(s.WithOpName("id1").WithControlDependencies(add), x); in TEST_F()
200 Output c1 = ops::Const(scope.WithOpName("c1").WithControlDependencies(neg), in TEST_F()
207 ops::Const(scope.WithOpName("c2").WithControlDependencies(ctrl_dep_id), in TEST_F()
246 ops::Identity(s.WithOpName("id1").WithControlDependencies(add), x); in TEST_F()
248 ops::Identity(s.WithOpName("id2").WithControlDependencies(add), y); in TEST_F()
266 auto noop2 = ops::NoOp(s.WithControlDependencies(x)); in TEST_F()
[all …]
|
| D | model_pruner_test.cc | 149 ops::IdentityN(s.WithOpName("e").WithControlDependencies(d), {a, b, c}); in TEST_F()
198 ops::IdentityN(s.WithOpName("e").WithControlDependencies(d), {a, b, c}); in TEST_F()
241 Output d = ops::AddN(s.WithOpName("d").WithControlDependencies(b), {c}); in TEST_F()
287 ops::Identity(scope.WithOpName("id1").WithControlDependencies(v_ctrl), in TEST_F()
291 scope.WithOpName("id3").WithControlDependencies(id0), id1); in TEST_F()
517 auto n2 = ops::NoOp(s.WithOpName("no_op2").WithControlDependencies(c1)); in TEST_F()
520 s.WithOpName("id1").WithControlDependencies({n1, n2}), c1); in TEST_F()
535 auto n2 = ops::NoOp(s.WithOpName("no_op2").WithControlDependencies(c1)); in TEST_F()
537 ops::Identity(s.WithOpName("id1").WithControlDependencies({n2}), c1); in TEST_F()
555 Output c2 = ops::Const(s.WithOpName("c2").WithControlDependencies({c0}), in TEST_F()
[all …]
|
| D | scoped_allocator_optimizer_test.cc | 178 Output a1 = ops::Abs(s.WithOpName("a1").WithControlDependencies({ctl1}), a); in BuildAbsGraphWithInputAndOutputControlEdges()
179 Output a2 = ops::Abs(s.WithOpName("a2").WithControlDependencies({ctl2}), b); in BuildAbsGraphWithInputAndOutputControlEdges()
183 ops::Const<float>(s.WithOpName("ctl3").WithControlDependencies({a1}), in BuildAbsGraphWithInputAndOutputControlEdges()
186 ops::Const<float>(s.WithOpName("ctl4").WithControlDependencies({a2}), in BuildAbsGraphWithInputAndOutputControlEdges()
|
| D | debug_stripper_test.cc | 57 s.WithOpName("add").WithControlDependencies({assert.operation}), x, y); in TEST_F()
84 s.WithOpName("z").WithControlDependencies({assert.operation}), x, y); in TEST_F()
|
| D | arithmetic_optimizer_test.cc | 145 Output div = ops::Div(s.WithOpName("div").WithControlDependencies( in TEST_F()
219 Output mul = ops::Mul(s.WithControlDependencies(d).WithOpName("mul"), c, c); in TEST_F()
345 s.WithOpName("recip2").WithControlDependencies(squeeze), c); in TEST_F()
420 Output add = ops::Add(s.WithOpName("add").WithControlDependencies(y), x, x); in TEST_F()
1490 ops::Identity(s.WithOpName("outputs").WithControlDependencies(transpose2), in TEST_F()
1551 s.WithOpName("transpose1").WithControlDependencies(perm2), inputs, perm1); in TEST_F()
2403 s.WithOpName("Neg_x_with_dep").WithControlDependencies({add_x_y}), x); in TEST_F()
2806 auto a12 = ops::Add(s.WithOpName("a12").WithControlDependencies(x3), x1, x2); in TEST_F()
2845 auto exp1 = ops::Exp(s.WithOpName("exp1").WithControlDependencies(x3), x1); in TEST_F()
3136 ops::Sin(s.WithOpName("sin_a").WithControlDependencies(ctrl3), a); in TEST_F()
[all …]
|
| D | constant_folding_test.cc | 1243 ops::Const(scope.WithOpName("c").WithControlDependencies(p1), 10, {3}); in TEST_F()
1246 ops::Identity(scope.WithOpName("i2").WithControlDependencies(p2), {i1}); in TEST_F()
1287 ops::Const(scope.WithOpName("c").WithControlDependencies(p1), 10, {3}); in TEST_F()
1290 ops::Identity(scope.WithOpName("i2").WithControlDependencies(p2), {i1}); in TEST_F()
1342 ops::Const(scope.WithOpName("c").WithControlDependencies(p1), 10, {3}); in TEST_F()
1344 .WithControlDependencies(p2) in TEST_F()
1345 .WithControlDependencies(p1), in TEST_F()
1897 ops::Const(scope.WithOpName("const1").WithControlDependencies(x), 2.7f, in TEST_F()
1902 ops::Const(scope.WithOpName("const3").WithControlDependencies(x), 3.14f, in TEST_F()
2099 ops::Transpose t2(scope.WithOpName("t2").WithControlDependencies({in1}), in2, in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | functionalize_control_flow_test.cc | 78 scope.WithOpName("cond").WithControlDependencies(identity_t), 17); in TEST()
86 scope.WithOpName("cond").WithControlDependencies(identity_f), 23); in TEST()
139 scope.WithOpName("cond").WithControlDependencies(identity), 17); in TEST()
164 scope.WithOpName("cond_1").WithControlDependencies(identity), 23); in TEST()
219 scope.WithOpName("while/Less/y").WithControlDependencies(merge.output), in TEST()
230 scope.WithOpName("while/add/y").WithControlDependencies(identity), 1); in TEST()
283 scope.WithOpName("while/Less/y").WithControlDependencies(arg), 10); in TEST()
305 scope.WithOpName("while/add/y").WithControlDependencies(identity), 1); in TEST()
369 scope.WithOpName("while/Less/y").WithControlDependencies(merge.output), in TEST()
499 scope.WithOpName("while/Less/y").WithControlDependencies(merge.output), in TEST()
[all …]
|
| D | const_analysis_test.cc | 98 ops::Const(root.WithOpName("c1").WithControlDependencies(arg0), 1, {1}); in TEST()
119 ops::Const(root.WithOpName("c1").WithControlDependencies(arg0), 1, {1}); in TEST()
144 ops::Const(root.WithOpName("c1").WithControlDependencies(arg0), 1, {1}); in TEST()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | graph_partition_test.cc | 276 Combine(in_.WithOpName("B2").WithControlDependencies(a1), b1, b1); in TEST_F()
284 auto c = Const(scope_a_.WithOpName("A1/_0").WithControlDependencies(a1), {}); in TEST_F()
292 Combine(scope_b_.WithOpName("B2").WithControlDependencies(id), b1, b1); in TEST_F()
322 Combine(in_.WithOpName("B2").WithControlDependencies(a1), b1, b1); in TEST_F()
323 FloatInput(in_.WithOpName("B3").WithControlDependencies(a1)); in TEST_F()
331 auto c = Const(scope_a_.WithOpName("A1/_0").WithControlDependencies(a1), {}); in TEST_F()
339 Combine(scope_b_.WithOpName("B2").WithControlDependencies(id), b1, b1); in TEST_F()
340 FloatInput(scope_b_.WithOpName("B3").WithControlDependencies(id)); in TEST_F()
348 FloatInput(in_.WithOpName("B3").WithControlDependencies(a1)); in TEST_F()
357 auto c = Const(scope_a_.WithOpName("A1/_2").WithControlDependencies(a1), {}); in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/cc/framework/ |
| D | scope.h | 128 Scope WithControlDependencies(
132 Scope WithControlDependencies(const Output& control_dep) const;
|
| D | scope_test.cc | 144 Scope c = root.WithControlDependencies({c1, c2}); in TEST()
146 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()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_test.cc | 829 .WithControlDependencies({c})); in TEST_F()
831 s.WithOpName("Func/b/input/_1").WithControlDependencies({func0}), a); in TEST_F()
853 .WithControlDependencies({c})); in TEST_F()
855 s.WithOpName("Func/b/input/_1").WithControlDependencies({func0}), a); in TEST_F()
858 .WithControlDependencies({func0})); in TEST_F()
860 s.WithOpName("Func/b/x2/input/_4").WithControlDependencies({func3}), in TEST_F()
863 s.WithOpName("b/x2/two").WithControlDependencies({func3}), 2LL); in TEST_F()
869 .WithControlDependencies({func0})); in TEST_F()
871 s.WithOpName("Func/b/y/input/_7").WithControlDependencies({func6}), in TEST_F()
874 s.WithOpName("b/y/two").WithControlDependencies({func6}), 2LL); in TEST_F()
[all …]
|
| D | constant_folding_test.cc | 413 auto c1 = ops::Const<int>(s.WithControlDependencies(recv1), 2); in TEST_F()
416 auto c2 = ops::Const<int>(s.WithControlDependencies(recv2), 3); in TEST_F()
417 auto add = ops::Add(s.WithControlDependencies(c2), c0, c1); in TEST_F()
457 auto c = ops::Const<int>(s.WithControlDependencies(recv2), 3); in TEST_F()
458 auto add0 = ops::Add(s.WithControlDependencies(c), rank, size); in TEST_F()
598 ops::Const<int>(s.WithOpName("c0").WithControlDependencies(recv0), 1); in TEST_F()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | graph_to_functiondef_test.cc | 99 auto b = ops::Neg(root.WithOpName("b").WithControlDependencies(a), a); in TEST()
100 auto c = ops::_Retval(root.WithOpName("c").WithControlDependencies(b), b, 0); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/fuzzing/ |
| D | decode_compressed_fuzz.cc | 36 scope.WithControlDependencies(std::vector<tensorflow::Operation>{ in BuildGraph()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 232 ops::Neg(s.WithControlDependencies(neg0).WithOpName("neg1"), c0); in TEST_F()
236 auto noop = ops::NoOp(s.WithControlDependencies(neg0).WithOpName("noop")); in TEST_F()
342 auto add = ops::Add(s.WithOpName("add").WithControlDependencies(mul), a, b); in TEST_F()
348 ops::Round(s.WithOpName("round").WithControlDependencies(add), mul); in TEST_F()
|
| D | graph_view_test.cc | 185 Output d = ops::AddN(s.WithOpName("d").WithControlDependencies(a), {b, c}); in TEST_F()
249 Output d = ops::AddN(s.WithOpName("d").WithControlDependencies(a), {b, c}); in TEST_F()
271 Output d = ops::AddN(s.WithOpName("d").WithControlDependencies(a), {b, c}); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | shape_inference_test.cc | 85 scope.WithOpName("while/Less/y").WithControlDependencies(merge.output), in TEST()
101 scope.WithOpName("while/add/y").WithControlDependencies(identity), 1); in TEST()
|
| /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/cc/gradients/ |
| D | math_grad.cc | 204 Scope grad_scope = scope.WithControlDependencies(grad); in TanhGrad()
260 Scope grad_scope = scope.WithControlDependencies(grad); in SigmoidGrad()
758 Scope grad_scope = scope.WithControlDependencies(grad); in ErfGrad()
775 Scope grad_scope = scope.WithControlDependencies(grad); in ErfinvGrad()
792 Scope grad_scope = scope.WithControlDependencies(grad); in NdtriGrad()
807 Scope grad_scope = scope.WithControlDependencies(grad); in LgammaGrad()
|
| /external/tensorflow/tensorflow/go/op/ |
| D | scope.go | 115 func (s *Scope) WithControlDependencies(ops ...*tf.Operation) *Scope { func
|
| D | scope_test.go | 83 s = s.WithControlDependencies(readDeps...)
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 725 auto b = ops::RandomUniform(s.WithOpName("b").WithControlDependencies(a), in CreateGrapplerItemWithMatmulChain()
727 auto c = ops::RandomUniform(s.WithOpName("c").WithControlDependencies(b), in CreateGrapplerItemWithMatmulChain()
729 auto d = ops::RandomUniform(s.WithOpName("d").WithControlDependencies(c), in CreateGrapplerItemWithMatmulChain()
731 auto e = ops::RandomUniform(s.WithOpName("e").WithControlDependencies(d), in CreateGrapplerItemWithMatmulChain()
734 auto ab = ops::MatMul(s.WithOpName("ab").WithControlDependencies(e), a, b); in CreateGrapplerItemWithMatmulChain()
795 ops::NoOp(s.WithControlDependencies(input_tensors).WithOpName("out")); in CreateGrapplerItemWithControlDependency()
881 auto z4 = ops::NoOp(s.WithControlDependencies(batch_var).WithOpName("z4")); in CreateGrapplerItemWithBatchNorm()
2169 .WithControlDependencies(y) in CreateGrapplerItemWithInterDeviceTransfers()
|
| D | graph_memory_test.cc | 202 s.WithOpName("init").WithDevice("/CPU:0").WithControlDependencies( in TEST_F()
|