Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for GetNodesRelatedToRefVariables (0.32 sec)

  1. tensorflow/compiler/jit/xla_cluster_util.cc

        // size of `callee_ref_nodes` but for now we don't ceare.
        if (!callee_ref_nodes.empty()) {
          return true;
        }
      }
    
      return false;
    }
    
    // Helper for GetNodesRelatedToRefVariables that traverses the graph in one
    // direction.
    Status GetNodesRelatedToRefVariablesInDirection(
        const Graph& graph, FunctionLibraryRuntime* lib_runtime,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 29 08:39:39 UTC 2024
    - 21.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/jit/xla_cluster_util_test.cc

      FunctionLibraryRuntime* lib_runtime =
          pflr->GetFLR(ProcessFunctionLibraryRuntime::kDefaultFLRDevice);
    
      TF_ASSIGN_OR_RETURN(absl::flat_hash_set<Node*> nodes_related_to_ref_vars,
                          GetNodesRelatedToRefVariables(*graph, lib_runtime));
    
      std::vector<string> names;
      absl::c_transform(nodes_related_to_ref_vars, std::back_inserter(names),
                        [](Node* n) { return n->name(); });
      absl::c_sort(names);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Feb 21 09:53:30 UTC 2024
    - 10.8K bytes
    - Viewed (0)
Back to top