Home
last modified time | relevance | path

Searched defs:visited_nodes (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dcheck_strict_fusion.cpp32 std::unordered_set<Node*> visited_nodes; in collectValuesUsedInGuard() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dmemory_optimizer.cc174 std::unordered_set<const NodeDef*> visited_nodes; in GetOpGroupsToRecompute() local
H A Dgeneric_layout_optimizer_transposer.cc1097 absl::flat_hash_set<utils::MutableNodeView*> visited_nodes; in IsAfterDstToSrcTransform() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_instruction_test.cc583 const std::vector<const HloInstruction*>& visited_nodes() { in visited_nodes() function in xla::__anon6a94cac00111::NodeCollectorAndPostProcessor