| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | dependency_optimizer_test.cc | 72 ops::Identity(s.WithOpName("id1").WithControlDependencies(x), add); in TEST_F()
74 s.WithOpName("id2").WithControlDependencies(y).WithControlDependencies(z), in TEST_F()
108 ops::Identity(s.WithOpName("id1").WithControlDependencies(add), x); in TEST_F()
110 ops::Identity(s.WithOpName("id2").WithControlDependencies(add), y); in TEST_F()
154 ops::Identity(s.WithOpName("id1").WithControlDependencies(add), x); in TEST_F()
199 Output c1 = ops::Const(scope.WithOpName("c1").WithControlDependencies(neg), in TEST_F()
206 ops::Const(scope.WithOpName("c2").WithControlDependencies(ctrl_dep_id), in TEST_F()
245 ops::Identity(s.WithOpName("id1").WithControlDependencies(add), x); in TEST_F()
247 ops::Identity(s.WithOpName("id2").WithControlDependencies(add), y); in TEST_F()
265 auto noop2 = ops::NoOp(s.WithControlDependencies(x)); in TEST_F()
[all …]
|
| D | model_pruner_test.cc | 155 ops::IdentityN(s.WithOpName("e").WithControlDependencies(d), {a, b, c}); in TEST_F()
215 ops::IdentityN(s.WithOpName("e").WithControlDependencies(d), {a, b, c}); in TEST_F()
261 Output d = ops::AddN(s.WithOpName("d").WithControlDependencies(b), {c}); in TEST_F()
309 scope.WithOpName("id1").WithControlDependencies(v_ctrl), s.output_false); in TEST_F()
312 ops::Identity(scope.WithOpName("id3").WithControlDependencies(id0), id1); in TEST_F()
|
| 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 | constant_folding_test.cc | 259 Output c1 = ops::Const(s.WithOpName("c1").WithControlDependencies(add_child), in TEST_F()
993 ops::Const(scope.WithOpName("c").WithControlDependencies(p1), 10, {3}); in TEST_F()
996 ops::Identity(scope.WithOpName("i2").WithControlDependencies(p2), {i1}); in TEST_F()
1037 ops::Const(scope.WithOpName("c").WithControlDependencies(p1), 10, {3}); in TEST_F()
1040 ops::Identity(scope.WithOpName("i2").WithControlDependencies(p2), {i1}); in TEST_F()
1092 ops::Const(scope.WithOpName("c").WithControlDependencies(p1), 10, {3}); in TEST_F()
1094 .WithControlDependencies(p2) in TEST_F()
1095 .WithControlDependencies(p1), in TEST_F()
1647 ops::Const(scope.WithOpName("const1").WithControlDependencies(x), 2.7f, in TEST_F()
1652 ops::Const(scope.WithOpName("const3").WithControlDependencies(x), 3.14f, in TEST_F()
[all …]
|
| D | arithmetic_optimizer_test.cc | 142 Output div = ops::Div(s.WithOpName("div").WithControlDependencies( in TEST_F()
217 Output mul = ops::Mul(s.WithControlDependencies(d).WithOpName("mul"), c, c); in TEST_F()
330 s.WithOpName("recip2").WithControlDependencies(squeeze), c); in TEST_F()
402 Output add = ops::Add(s.WithOpName("add").WithControlDependencies(y), x, x); in TEST_F()
1439 ops::Identity(s.WithOpName("outputs").WithControlDependencies(transpose2), in TEST_F()
1500 s.WithOpName("transpose1").WithControlDependencies(perm2), inputs, perm1); in TEST_F()
2330 s.WithOpName("Neg_x_with_dep").WithControlDependencies({add_x_y}), x); in TEST_F()
2694 auto a12 = ops::Add(s.WithOpName("a12").WithControlDependencies(x3), x1, x2); in TEST_F()
2736 auto exp1 = ops::Exp(s.WithOpName("exp1").WithControlDependencies(x3), x1); in TEST_F()
2980 ops::Sin(s.WithOpName("sin_a").WithControlDependencies(ctrl3), a); 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()
138 scope.WithOpName("cond").WithControlDependencies(identity), 17); in TEST()
163 scope.WithOpName("cond_1").WithControlDependencies(identity), 23); in TEST()
218 scope.WithOpName("while/Less/y").WithControlDependencies(merge.output), in TEST()
229 scope.WithOpName("while/add/y").WithControlDependencies(identity), 1); in TEST()
282 scope.WithOpName("while/Less/y").WithControlDependencies(arg), 10); in TEST()
304 scope.WithOpName("while/add/y").WithControlDependencies(identity), 1); in TEST()
368 scope.WithOpName("while/Less/y").WithControlDependencies(merge.output), in TEST()
502 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()
|
| D | xla_compiler_test.cc | 292 scope.WithControlDependencies(std::vector<Operation>{write}), var, in TEST_F()
342 scope.WithControlDependencies(std::vector<Operation>{write}), var, in TEST_F()
386 scope.WithControlDependencies(std::vector<Operation>{write}), var, in TEST_F()
990 scope.WithControlDependencies(std::vector<Operation>{write}), var, in TEST_F()
1142 scope.WithControlDependencies(std::vector<Operation>{write}), var, in TEST_F()
1178 scope.WithControlDependencies(std::vector<Operation>{write}), var, in BuildTestGraph()
|
| /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/core/common_runtime/ |
| D | function_test.cc | 826 .WithControlDependencies({c})); in TEST_F()
828 s.WithOpName("Func/b/input/_1").WithControlDependencies({func0}), a); in TEST_F()
850 .WithControlDependencies({c})); in TEST_F()
852 s.WithOpName("Func/b/input/_1").WithControlDependencies({func0}), a); in TEST_F()
855 .WithControlDependencies({func0})); in TEST_F()
857 s.WithOpName("Func/b/x2/input/_4").WithControlDependencies({func3}), in TEST_F()
860 s.WithOpName("b/x2/two").WithControlDependencies({func3}), 2LL); in TEST_F()
866 .WithControlDependencies({func0})); in TEST_F()
868 s.WithOpName("Func/b/y/input/_7").WithControlDependencies({func6}), in TEST_F()
871 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/cc/framework/ |
| D | scope.h | 128 Scope WithControlDependencies(
132 Scope WithControlDependencies(const Output& control_dep) const;
|
| D | scope_test.cc | 142 Scope c = root.WithControlDependencies({c1, c2}); in TEST()
144 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/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/compiler/jit/ |
| D | shape_inference_test.cc | 83 scope.WithOpName("while/Less/y").WithControlDependencies(merge.output), in TEST()
99 scope.WithOpName("while/add/y").WithControlDependencies(identity), 1); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 230 ops::Neg(s.WithControlDependencies(neg0).WithOpName("neg1"), c0); in TEST_F()
234 auto noop = ops::NoOp(s.WithControlDependencies(neg0).WithOpName("noop")); in TEST_F()
340 auto add = ops::Add(s.WithOpName("add").WithControlDependencies(mul), a, b); in TEST_F()
346 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/core/kernels/fuzzing/ |
| D | decode_compressed_fuzz.cc | 36 scope.WithControlDependencies(std::vector<tensorflow::Operation>{ in BuildGraph()
|
| /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 | 167 auto b = ops::RandomUniform(s.WithOpName("b").WithControlDependencies(a), in CreateGrapplerItemWithMatmulChain()
169 auto c = ops::RandomUniform(s.WithOpName("c").WithControlDependencies(b), in CreateGrapplerItemWithMatmulChain()
171 auto d = ops::RandomUniform(s.WithOpName("d").WithControlDependencies(c), in CreateGrapplerItemWithMatmulChain()
173 auto e = ops::RandomUniform(s.WithOpName("e").WithControlDependencies(d), in CreateGrapplerItemWithMatmulChain()
176 auto ab = ops::MatMul(s.WithOpName("ab").WithControlDependencies(e), a, b); in CreateGrapplerItemWithMatmulChain()
244 ops::NoOp(s.WithControlDependencies(input_tensors).WithOpName("out")); in CreateGrapplerItemWithControlDependency()
285 auto z4 = ops::NoOp(s.WithControlDependencies(batch_var).WithOpName("z4")); in CreateGrapplerItemWithBatchNorm()
1574 .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 | 115 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 | 205 Scope grad_scope = scope.WithControlDependencies(grad); in TanhGrad()
261 Scope grad_scope = scope.WithControlDependencies(grad); in SigmoidGrad()
759 Scope grad_scope = scope.WithControlDependencies(grad); in ErfGrad()
774 Scope grad_scope = scope.WithControlDependencies(grad); in LgammaGrad()
|