- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for assigned_device (0.15 sec)
-
tensorflow/compiler/jit/node_matchers.cc
<< node->type_string(); } return false; } if (assigned_device && node->assigned_device_name() != *assigned_device) { if (listener->IsInterested()) { *listener << "\nexpected assigned_device " << *assigned_device << " but found \"" << node->assigned_device_name() << "\""; } return false; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 03 16:15:20 UTC 2022 - 16.8K bytes - Viewed (0) -
tensorflow/cc/framework/scope.cc
Scope Scope::WithDevice(const string& device) const { return Scope(new Impl(*this, Impl::Tags::Device(), device)); } Scope Scope::WithAssignedDevice(const string& assigned_device) const { return Scope(new Impl(*this, Impl::Tags::AssignedDevice(), assigned_device)); } Scope Scope::WithXlaCluster(const string& xla_cluster) const { return Scope(new Impl(*this, Impl::Tags::XlaCluster(), xla_cluster)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/cc/framework/scope.h
Scope WithDevice(const string& device) const; /// Returns a new scope. All ops created within the returned scope will have /// their assigned device set to `assigned_device`. Scope WithAssignedDevice(const string& assigned_device) const; /// Returns a new scope. All ops created within the returned scope will have /// their _XlaCluster attribute set to `xla_cluster`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 09:08:33 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/jit/increase_dynamism_for_auto_jit_pass_test.cc
auto m_slice_size_0 = Out(NodeWith( Op("Sub"), AssignedDevice(kHostName), Inputs( Out(NodeWith(Op("Slice"), AssignedDevice(kHostName), Inputs(m_input_shape, Const(zero_64), Const(one_64)))), Out(NodeWith(Op("Slice"), AssignedDevice(kHostName), Inputs(m_begin_s64, Const(zero_64), Const(one_64)))))));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 18.3K bytes - Viewed (0)