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