Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for assigned_device (0.15 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
Back to top