Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for GetNodeAttr (0.22 sec)

  1. tensorflow/compiler/jit/cluster_scoping_pass.cc

     private:
      Graph* graph_;
      OptimizerOptions::GlobalJitLevel global_jit_level_;
      size_t unique_scope_id_;
    };
    
    std::optional<string> GetXlaInternalScope(Node* node) {
      string scope;
      if (GetNodeAttr(node->attrs(), kXlaInternalScopeAttr, &scope).ok()) {
        return scope;
      }
    
      return std::nullopt;
    }
    
    void SetXlaInternalScope(Node* node, StringPiece scope) {
      node->AddAttr(kXlaInternalScopeAttr, scope);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 5.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/clone_constants_for_better_clustering.cc

      }
    
      new_in->set_assigned_device_name(n->assigned_device_name());
      return new_in;
    }
    
    namespace {
    absl::StatusOr<bool> IsConstantSmall(Node* n) {
      const TensorProto* proto = nullptr;
      TF_RETURN_IF_ERROR(GetNodeAttr(n->def(), "value", &proto));
    
      int64_t total_elements = 1;
      for (const auto& dim : proto->tensor_shape().dim()) {
        if (dim.size() < 0) {
          return errors::Internal("Unknown dimension size in constant tensor ",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 12 06:33:33 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  3. tensorflow/cc/framework/cc_ops_test.cc

      return BiasAdd(cop_scopes.last, m, b);
    }
    
    void GetColocationConstraints(const Output& tensor,
                                  std::vector<string>* constraints) {
      constraints->clear();
      TF_EXPECT_OK(GetNodeAttr(tensor.op().node()->attrs(), kColocationAttrName,
                               constraints));
    }
    
    TEST(CCOpTest, Basic) {
      Scope root = Scope::NewRootScope();
      auto c = Const(root, {{1, 1}});
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 15 15:13:38 UTC 2023
    - 8.7K bytes
    - Viewed (0)
Back to top