Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for assigned_device (0.17 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/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)
  3. 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)
Back to top