Home
last modified time | relevance | path

Searched refs:WithDevice (Results 1 – 25 of 39) sorted by relevance

12

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer_test.cc232 ops::Variable(s.WithOpName("a").WithDevice("/gpu:0"), {10, 10}, DT_FLOAT); in TEST_F()
233 Output b = ops::AddN(s.WithOpName("b").WithDevice("/gpu:0"), {a}); in TEST_F()
234 Output c = ops::AddN(s.WithOpName("c").WithDevice("/gpu:0"), {b}); in TEST_F()
235 Output d = ops::AddN(s.WithOpName("d").WithDevice("/gpu:0"), {c}); in TEST_F()
236 Output e = ops::AddN(s.WithOpName("e").WithDevice("/gpu:0"), {b, d}); in TEST_F()
298 Output v = ops::Variable(s.WithOpName("v").WithDevice("/gpu:0"), in TEST_F()
300 Output a = ops::Identity(s.WithOpName("a").WithDevice("/gpu:0"), v); in TEST_F()
301 Output b = ops::Square(s.WithOpName("b").WithDevice("/gpu:0"), v); in TEST_F()
302 Output c = ops::Sqrt(s.WithOpName("c").WithDevice("/gpu:0"), a); in TEST_F()
303 Output d = ops::Identity(s.WithOpName("d").WithDevice("/gpu:0"), b); in TEST_F()
[all …]
Ddependency_optimizer_test.cc294 Output x = ops::RandomUniform(s.WithOpName("x").WithDevice("/CPU:0"), {1, 2}, in TEST_F()
296 Output y = ops::RandomUniform(s.WithOpName("y").WithDevice("/CPU:0"), {1, 2}, in TEST_F()
299 auto noop = ops::NoOp(s.WithControlDependencies(x).WithDevice("/CPU:1")); in TEST_F()
302 s.WithControlDependencies(x).WithControlDependencies(y).WithDevice( in TEST_F()
305 s.WithControlDependencies({noop.operation}).WithDevice("/CPU:1"), x); in TEST_F()
308 .WithDevice("/CPU:1"), in TEST_F()
329 Output x = ops::RandomUniform(s.WithOpName("x").WithDevice("/CPU:0"), {1, 2}, in TEST_F()
331 Output y = ops::RandomUniform(s.WithOpName("y").WithDevice("/CPU:0"), {1, 2}, in TEST_F()
334 auto id_a = ops::Identity(s.WithOpName("id_a").WithDevice("/CPU:1"), x); in TEST_F()
337 s.WithOpName("id_b").WithControlDependencies(y).WithDevice("/CPU:0"), x); in TEST_F()
[all …]
Dmodel_pruner_test.cc473 Output c = ops::Const(s.WithOpName("c").WithDevice("/cpu:0"), 0.0f, {10, 10}); in TEST_F()
476 Output i1 = ops::Identity(s.WithOpName("i1").WithDevice("/device:GPU:0"), c); in TEST_F()
477 Output a1 = ops::Sqrt(s.WithOpName("a1").WithDevice("/device:GPU:0"), {i1}); in TEST_F()
478 Output a2 = ops::Sqrt(s.WithOpName("a2").WithDevice("/device:GPU:0"), {i1}); in TEST_F()
481 Output i2 = ops::Identity(s.WithOpName("i2").WithDevice("/cpu:0"), c); in TEST_F()
482 Output a3 = ops::Sqrt(s.WithOpName("a3").WithDevice("/device:GPU:0"), {i2}); in TEST_F()
483 Output a4 = ops::Sqrt(s.WithOpName("a4").WithDevice("/device:GPU:0"), {i2}); in TEST_F()
Dpin_to_host_optimizer_test.cc184 ops::Const(s.WithOpName("a").WithDevice("/device:GPU:0"), 1, {64, 64}); in TEST_F()
187 ops::ReduceProd(s.WithOpName("c").WithDevice("/device:GPU:0"), a, b); in TEST_F()
188 Output d = ops::Identity(s.WithDevice("/device:CPU:0").WithOpName("d"), c); in TEST_F()
Dscoped_allocator_optimizer_test.cc80 s = s.WithDevice("/job:localhost/replica:0/task:0/device:CPU:0"); in BuildAbsGraph()
/external/tensorflow/tensorflow/compiler/xrt/tests/
Draw_api_test.cc252 Scope root = Scope::NewRootScope().WithDevice(DeviceFromFlag()); in TEST()
283 Scope root = Scope::NewRootScope().WithDevice(DeviceFromFlag()); in TEST()
310 Scope root = Scope::NewRootScope().WithDevice(DeviceFromFlag()); in TEST()
335 Scope root = Scope::NewRootScope().WithDevice(DeviceFromFlag()); in TEST()
365 Scope root = Scope::NewRootScope().WithDevice(DeviceFromFlag()); in TEST()
367 ops::Const(root.WithDevice("/device:CPU:0"), alloc.SerializeAsString()); in TEST()
385 auto new_value = ops::Const(root.WithDevice("/device:CPU:0"), in TEST()
419 Scope root = Scope::NewRootScope().WithDevice(DeviceFromFlag()); in TEST()
421 ops::Const(root.WithDevice("/device:CPU:0"), alloc1.SerializeAsString()); in TEST()
423 ops::Const(root.WithDevice("/device:CPU:0"), alloc2.SerializeAsString()); in TEST()
[all …]
/external/tensorflow/tensorflow/core/grappler/inputs/
Dtrivial_test_graph_input_yielder.cc40 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/graph/
Dcollective_order_test.cc86 .WithDevice(device) in CollectiveReduceNode()
115 builder.opts().WithName("a").WithDevice(dev0)); in InitGraph()
117 builder.opts().WithName("b").WithDevice(dev1)); in InitGraph()
122 builder.opts().WithName("id0").WithDevice(dev0).WithAttr("T", DT_FLOAT)); in InitGraph()
125 builder.opts().WithName("id1").WithDevice(dev1).WithAttr("T", DT_FLOAT)); in InitGraph()
170 builder.opts().WithName("a").WithDevice(dev0)); in InitGraph2()
175 builder.opts().WithName("id").WithDevice(dev0).WithAttr("T", DT_FLOAT)); in InitGraph2()
206 builder.opts().WithName("w").WithDevice(dev0)); in InitGraphForPruning()
208 builder.opts().WithName("x").WithDevice(dev0)); in InitGraphForPruning()
210 builder.opts().WithName("y").WithDevice(dev0)); in InitGraphForPruning()
[all …]
Dgraph_partition_test.cc200 scope_a_(Scope::NewRootScope().ExitOnError().WithDevice( in GraphPartitionTest()
202 scope_b_(Scope::NewRootScope().ExitOnError().WithDevice( in GraphPartitionTest()
412 Scope cpu0 = in_.WithDevice("/job:a/replica:0/task:0/cpu:0"); in TEST_F()
425 Scope cpu1 = s.WithDevice("/job:a/replica:0/task:0/cpu:1"); in TEST_F()
Dgraph_def_builder.h80 Options WithDevice(StringPiece device) const;
Dgraph_def_builder.cc33 GraphDefBuilder::Options GraphDefBuilder::Options::WithDevice( in WithDevice() function in tensorflow::GraphDefBuilder::Options
/external/tensorflow/tensorflow/compiler/jit/
Dclone_constants_for_better_clustering_test.cc65 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST()
66 Scope on_cpu = root.WithAssignedDevice(kCPU).WithDevice(kCPU); in TEST()
92 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST()
118 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST()
119 Scope on_cpu = root.WithAssignedDevice(kCPU).WithDevice(kCPU); in TEST()
147 Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); in TEST()
148 Scope on_cpu = root.WithAssignedDevice(kCPU).WithDevice(kCPU); in TEST()
Dbuild_xla_ops_pass_test.cc134 Scope root = Scope::NewRootScope().WithDevice(kXlaDeviceName).ExitOnError(); in TEST_F()
208 Scope root = Scope::NewRootScope().WithDevice(kXlaDeviceName).ExitOnError(); in TEST_F()
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_copy_test.cc165 Scope root = Scope::NewRootScope().WithDevice("/cpu:0"); in TEST()
199 Scope root = Scope::NewRootScope().WithDevice("/gpu:0"); in TEST()
242 Scope root = Scope::NewRootScope().WithDevice("/gpu:0"); in TEST()
269 Scope root = Scope::NewRootScope().WithDevice("/cpu:0"); in TEST()
293 Scope root = Scope::NewRootScope().WithDevice("/cpu:0"); in TEST()
319 Scope root = Scope::NewRootScope().WithDevice("/cpu:0"); in TEST()
320 Scope with_gpu = root.WithDevice("/gpu:0"); in TEST()
356 Scope root = Scope::NewRootScope().WithDevice("/cpu:0"); in TEST()
357 Scope with_gpu = root.WithDevice("/gpu:0"); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_test.cc693 b.opts().WithName("in").WithDevice("/device:fakecpu:1")); in TEST_F()
715 ops::SourceOp("TestInput", b.opts().WithName("in").WithDevice("/job:a")); in TEST_F()
717 b.opts().WithName("var").WithDevice("/job:a")); in TEST_F()
752 b.opts().WithName("in").WithDevice("/device:fakegpu:0")); in TEST_F()
754 b.opts().WithName("var").WithDevice("/device:fakegpu:0")); in TEST_F()
961 b.opts().WithName("in").WithDevice("/job:a/task:0/device:fakegpu:0")); in TEST_F()
963 ops::SourceOp("TestVariable", b.opts().WithName("var_0").WithDevice( in TEST_F()
969 b.opts().WithName("assign").WithDevice( in TEST_F()
990 b.opts().WithName("in").WithDevice("/job:a/task:0")); in TEST_F()
994 "TestVariable", b.opts().WithName("var_0").WithDevice("/job:a/task:0")); in TEST_F()
[all …]
Dpartitioning_utils_test.cc60 s = s.WithDevice(device0_->name()); in SwapGraph()
78 Scope s1 = s.WithDevice("/job:a/replica:0/task:0/device:CPU:0"); in TwoDeviceSwapGraph()
79 Scope s2 = s.WithDevice("/job:a/replica:0/task:0/device:CPU:1"); in TwoDeviceSwapGraph()
96 Scope s1 = s.WithDevice("/job:a/replica:0/task:0/device:CPU:0"); in SubGraph()
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory_test.cc196 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()
Dvirtual_scheduler_test.cc117 Scope s = Scope::NewRootScope().WithDevice(kCPU0); in CreateGrapplerItemWithConv2Ds()
144 Scope s = Scope::NewRootScope().WithDevice(kCPU0); in CreateGrapplerItemWithConv2DAndVariable()
163 Scope s = Scope::NewRootScope().WithDevice(kCPU0); in CreateGrapplerItemWithMatmulChain()
197 Scope s = Scope::NewRootScope().WithDevice(kCPU0); in CreateGrapplerItemWithAddN()
217 Scope s = Scope::NewRootScope().WithDevice(kCPU0); in CreateGrapplerItemWithUnnecessaryPlaceholderNodes()
236 Scope s = Scope::NewRootScope().WithDevice(kCPU0); in CreateGrapplerItemWithControlDependency()
259 Scope s = Scope::NewRootScope().WithDevice(kCPU0); in CreateGrapplerItemWithBatchNorm()
1545 tensorflow::Scope s = tensorflow::Scope::NewRootScope().WithDevice(kCPU0); in CreateGrapplerItemWithInterDeviceTransfers()
1564 auto y1 = ops::Identity(s.WithOpName("y1").WithDevice(kCPU1), y); in CreateGrapplerItemWithInterDeviceTransfers()
1565 auto y2 = ops::Identity(s.WithOpName("y2").WithDevice(kCPU1), y); in CreateGrapplerItemWithInterDeviceTransfers()
[all …]
/external/tensorflow/tensorflow/cc/framework/
Dwhile_gradients_test.cc192 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()
Dscope.h140 Scope WithDevice(const string& device) const;
/external/tensorflow/tensorflow/core/util/
Dequal_graph_def_test.cc136 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/compiler/tf2xla/
Dfused_batchnorm_reserve_space_test.cc101 root.WithOpName("fused_batch_norm_tf").WithDevice(tf_device), input, in TEST()
104 root.WithOpName("fused_batch_norm_xla").WithDevice(xla_device), input, in TEST()
/external/tensorflow/tensorflow/go/op/
Dscope_test.go118 s = s.WithDevice("/device:GPU:0")
120 s = s.WithDevice("")
Dscope.go142 func (s *Scope) WithDevice(device string) *Scope { func
/external/tensorflow/tensorflow/cc/gradients/
Dmath_grad_test.cc59 CWiseUnaryGradTest() : scope_(Scope::NewRootScope().WithDevice("/cpu:0")) {} in CWiseUnaryGradTest()
572 MathGradTest() : root_(Scope::NewRootScope().WithDevice("/cpu:0")) {} in MathGradTest()
711 NaryGradTest() : scope_(Scope::NewRootScope().WithDevice("/cpu:0")) {} in NaryGradTest()

12