| /external/tensorflow/tensorflow/core/grappler/inputs/ |
| trivial_test_graph_input_yielder.cc | 40 Output x = RandomNormal(s.WithOpName("x").WithDevice("/CPU:0"),
51 s.WithDevice(
58 AddN(s.WithDevice(
69 FIFOQueue queue(s.WithOpName("queue").WithDevice("/CPU:0"),
71 QueueEnqueue enqueue(s.WithOpName("enqueue").WithDevice("/CPU:0"), queue,
73 QueueDequeue dequeue(s.WithOpName("dequeue").WithDevice("/CPU:0"), queue,
75 QueueClose cancel(s.WithOpName("cancel").WithDevice("/CPU:0"), queue,
81 AddN output(s.WithOpName("y").WithDevice("/CPU:0"), last_stage);
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| memory_optimizer_test.cc | 276 Output v = ops::Variable(s.WithOpName("v").WithDevice("/gpu:0"),
278 Output a = ops::Identity(s.WithOpName("a").WithDevice("/gpu:0"), v);
279 Output b = ops::Square(s.WithOpName("b").WithDevice("/gpu:0"), v);
280 Output c = ops::Sqrt(s.WithOpName("c").WithDevice("/gpu:0"), a);
281 Output d = ops::Identity(s.WithOpName("d").WithDevice("/gpu:0"), b);
284 ops::Concat(s.WithOpName("e").WithDevice("/gpu:0"), {a, b, c, d}, axis);
285 Output f = ops::Square(s.WithOpName("f").WithDevice("/gpu:0"), a);
286 Output g = ops::Sqrt(s.WithOpName("g").WithDevice("/gpu:0"), b);
287 Output h = ops::Exp(s.WithOpName("h").WithDevice("/gpu:0"), c);
288 Output i = ops::Log(s.WithOpName("i").WithDevice("/gpu:0"), d)
[all...] |
| model_pruner_test.cc | 314 Output c = ops::Const(s.WithOpName("c").WithDevice("/cpu:0"), 0.0f, {10, 10});
317 Output i1 = ops::Identity(s.WithOpName("i1").WithDevice("/device:GPU:0"), c);
318 Output a1 = ops::Sqrt(s.WithOpName("a1").WithDevice("/device:GPU:0"), {i1});
319 Output a2 = ops::Sqrt(s.WithOpName("a2").WithDevice("/device:GPU:0"), {i1});
322 Output i2 = ops::Identity(s.WithOpName("i2").WithDevice("/cpu:0"), c);
323 Output a3 = ops::Sqrt(s.WithOpName("a3").WithDevice("/device:GPU:0"), {i2});
324 Output a4 = ops::Sqrt(s.WithOpName("a4").WithDevice("/device:GPU:0"), {i2});
|
| dependency_optimizer_test.cc | 255 Output x = ops::RandomUniform(s.WithOpName("x").WithDevice("/CPU:0"), {1, 2},
257 Output y = ops::RandomUniform(s.WithOpName("y").WithDevice("/CPU:0"), {1, 2},
260 auto noop = ops::NoOp(s.WithControlDependencies(x).WithDevice("/CPU:1"));
263 s.WithControlDependencies(x).WithControlDependencies(y).WithDevice(
266 s.WithControlDependencies({noop.operation}).WithDevice("/CPU:1"), x);
269 .WithDevice("/CPU:1"),
|
| arithmetic_optimizer_test.cc | 718 tensorflow::Scope s = tensorflow::Scope::NewRootScope().WithDevice("/gpu:0");
754 tensorflow::Scope s = tensorflow::Scope::NewRootScope().WithDevice("/gpu:0");
[all...] |
| layout_optimizer_test.cc | 68 Output conv = ops::Conv2D(s->WithOpName("Conv2D").WithDevice(device), input,
[all...] |
| constant_folding_test.cc | 39 Output c = ops::AddN(s.WithOpName("c").WithDevice("/CPU:0"), {a, b});
[all...] |
| /external/tensorflow/tensorflow/core/framework/ |
| variant_op_copy_test.cc | 164 Scope root = Scope::NewRootScope().WithDevice("/cpu:0");
198 Scope root = Scope::NewRootScope().WithDevice("/gpu:0");
241 Scope root = Scope::NewRootScope().WithDevice("/gpu:0");
268 Scope root = Scope::NewRootScope().WithDevice("/cpu:0");
292 Scope root = Scope::NewRootScope().WithDevice("/cpu:0");
318 Scope root = Scope::NewRootScope().WithDevice("/cpu:0");
319 Scope with_gpu = root.WithDevice("/gpu:0");
355 Scope root = Scope::NewRootScope().WithDevice("/cpu:0");
356 Scope with_gpu = root.WithDevice("/gpu:0");
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| placer_test.cc | 402 b.opts().WithName("in").WithDevice("/device:fakecpu:1"));
424 ops::SourceOp("TestInput", b.opts().WithName("in").WithDevice("/job:a"));
426 b.opts().WithName("var").WithDevice("/job:a"));
461 b.opts().WithName("in").WithDevice("/device:fakegpu:0"));
463 b.opts().WithName("var").WithDevice("/device:fakegpu:0"));
622 b.opts().WithName("in").WithDevice("/job:a/task:0/device:fakegpu:0"));
624 ops::SourceOp("TestVariable", b.opts().WithName("var_0").WithDevice(
630 b.opts().WithName("assign").WithDevice(
652 b.opts().WithName("in").WithDevice("/job:a/task:0"));
656 "TestVariable", b.opts().WithName("var_0").WithDevice("/job:a/task:0"))
[all...] |
| function_test.cc | 791 .WithDevice("/job:localhost/replica:0/task:0/device:CPU:0"),
[all...] |
| /external/tensorflow/tensorflow/cc/framework/ |
| scope.h | 134 Scope WithDevice(const string& device) const;
|
| while_gradients_test.cc | 192 scope_ = scope_.WithDevice("/cpu:0");
204 Scope cpu1_scope = s.WithDevice("/cpu:1");
211 Scope cpu1_scope = scope_.WithDevice("/cpu:1");
|
| scope.cc | 382 Scope Scope::WithDevice(const string& device) const {
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| graph_memory_test.cc | 196 Output a = ops::Const(s.WithOpName("a").WithDevice("/CPU:0"), 10.0f, {3});
198 ops::Variable(s.WithOpName("v").WithDevice("/CPU:0"), {3}, DT_FLOAT);
200 ops::Assign(s.WithOpName("assign").WithDevice("/CPU:0"), v, a);
202 s.WithOpName("init").WithDevice("/CPU:0").WithControlDependencies(
|
| virtual_scheduler_test.cc | 110 Scope s = Scope::NewRootScope().WithDevice(kCPU0);
137 Scope s = Scope::NewRootScope().WithDevice(kCPU0);
156 Scope s = Scope::NewRootScope().WithDevice(kCPU0);
190 Scope s = Scope::NewRootScope().WithDevice(kCPU0);
210 Scope s = Scope::NewRootScope().WithDevice(kCPU0);
233 Scope s = Scope::NewRootScope().WithDevice(kCPU0);
794 tensorflow::Scope s = tensorflow::Scope::NewRootScope().WithDevice(kCPU0);
813 auto y1 = ops::Identity(s.WithOpName("y1").WithDevice(kCPU1), y);
814 auto y2 = ops::Identity(s.WithOpName("y2").WithDevice(kCPU1), y);
818 s.WithOpName("batch_mean1").WithDevice(kCPU1), batch_mean)
[all...] |
| /external/tensorflow/tensorflow/core/graph/ |
| graph_def_builder.h | 80 Options WithDevice(StringPiece device) const;
|
| graph_def_builder.cc | 33 GraphDefBuilder::Options GraphDefBuilder::Options::WithDevice(
|
| graph_partition_test.cc | 199 scope_a_(Scope::NewRootScope().ExitOnError().WithDevice(
201 scope_b_(Scope::NewRootScope().ExitOnError().WithDevice(
411 Scope cpu0 = in_.WithDevice("/job:a/replica:0/task:0/cpu:0");
424 Scope cpu1 = s.WithDevice("/job:a/replica:0/task:0/cpu:1");
|
| /external/tensorflow/tensorflow/core/kernels/ |
| shape_op_test.cc | 73 Scope root = Scope::NewRootScope().WithDevice("/cpu:0");
|
| /external/tensorflow/tensorflow/core/util/ |
| equal_graph_def_test.cc | 136 Input(e_.opts().WithName("A").WithDevice("/cpu:0"));
137 Input(a_.opts().WithName("A").WithDevice("/cpu:0"));
142 Input(e_.opts().WithName("A").WithDevice("/cpu:0"));
143 Input(a_.opts().WithName("A").WithDevice("/cpu:1"));
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| rpcbench_test.cc | 142 s.WithDevice(cluster->devices[use_multiple_devices ? j : 0].name()),
|
| /external/tensorflow/tensorflow/cc/gradients/ |
| math_grad_test.cc | 57 CWiseUnaryGradTest() : scope_(Scope::NewRootScope().WithDevice("/cpu:0")) {}
574 MathGradTest() : root_(Scope::NewRootScope().WithDevice("/cpu:0")) {}
713 NaryGradTest() : scope_(Scope::NewRootScope().WithDevice("/cpu:0")) {}
|
| math_grad.cc | [all...] |
| /external/tensorflow/tensorflow/compiler/jit/ |
| encapsulate_subgraphs_pass_test.cc | 600 Scope root = Scope::NewRootScope().ExitOnError().WithDevice(
631 Scope root = Scope::NewRootScope().ExitOnError().WithDevice(
689 Scope root = Scope::NewRootScope().ExitOnError().WithDevice(
729 Scope root = Scope::NewRootScope().ExitOnError().WithDevice(
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| transform_utils_test.cc | [all...] |
