Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for assigned_device (0.22 sec)

  1. tensorflow/compiler/jit/node_matchers.h

        name_ = std::move(name);
      }
    
      void set_op(string op) {
        DCHECK(IsEmpty());
        op_ = std::move(op);
      }
    
      void set_assigned_device(string assigned_device) {
        DCHECK(IsEmpty());
        assigned_device_ = std::move(assigned_device);
      }
    
      void set_constant_value(Tensor constant_value) {
        DCHECK(IsEmpty());
        constant_value_ = std::move(constant_value);
        op_ = "Const";
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Sep 06 19:12:29 UTC 2023
    - 8.1K bytes
    - Viewed (0)
  2. 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)
  3. 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)
  4. tensorflow/cc/framework/scope_internal.h

      Impl(const Scope& other, Tags::Colocate, const Operation& colocate_with_op,
           bool clear_colocations);
      Impl(const Scope& other, Tags::AssignedDevice, const string& assigned_device);
      Impl(const Scope& other, Tags::XlaCluster, const string& xla_cluster);
    
      std::unordered_set<string> GetColocationConstraints(
          const Operation& colocate_with_op) const;
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Oct 05 15:46:43 UTC 2022
    - 5.1K bytes
    - Viewed (0)
  5. 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)
  6. tensorflow/compiler/jit/node_matchers_test.cc

          assigned_add.node(),
          NodeWith(AssignedDevice("/job:localhost/replica:0/task:0/device:CPU:0")));
      EXPECT_THAT(unassigned_add.node(), NodeWith(AssignedDevice("")));
    
      EXPECT_EQ(Explain(unassigned_add.node(),
                        NodeWith(AssignedDevice(
                            "/job:localhost/replica:0/task:0/device:CPU:0"))),
                "\nexpected assigned_device "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 20 14:43:57 UTC 2024
    - 9.1K bytes
    - Viewed (0)
  7. 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