- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for WithAssignedDevice (0.11 sec)
-
tensorflow/compiler/jit/clone_constants_for_better_clustering_test.cc
} TEST(CloneConstantsForBetterClusteringTest, HostConstantPlacedOnCpu) { Scope root = Scope::NewRootScope().ExitOnError(); Scope on_gpu = root.WithAssignedDevice(kGPU).WithDevice(kGPU); Scope on_cpu = root.WithAssignedDevice(kCPU).WithDevice(kCPU); Output in0 = ops::Placeholder(on_gpu.WithOpName("in0"), DT_FLOAT); Output in1 = ops::Placeholder(on_gpu.WithOpName("in1"), DT_FLOAT);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 8.4K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_activity_listener_test.cc
} TestListener* listener() const { return listener_; } private: TestListener* listener_; }; GraphDef CreateGraphDef() { Scope root = Scope::NewRootScope().ExitOnError().WithAssignedDevice( "/job:localhost/replica:0/task:0/device:CPU:0"); Output a = ops::Placeholder(root.WithOpName("A"), DT_FLOAT); for (int i = 0; i < 5; i++) { a = ops::MatMul(root.WithOpName(absl::StrCat("matmul_", i)), a, a);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 5.9K bytes - Viewed (0)