Home
last modified time | relevance | path

Searched defs:output_dependencies (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_configuration_rewrite_pass.cc140 const std::vector<DistributedTPURewriteHelpers::OutputDependency>& in AddSynchronizationNode()
173 const std::vector<DistributedTPURewriteHelpers::OutputDependency>& in AddShutdownNode()
247 Graph* graph) -> Status { in Run()
365 Graph* graph) -> Status { in Run()
H A Ddistributed_tpu_rewrite_helpers.cc207 const std::vector<OutputDependency>& output_dependencies, in ForConfigurationNodeMatchingType()
234 std::vector<OutputDependency> output_dependencies; in ForConfigurationNodeMatchingType() local
H A Dconfigure_tpu_embedding_rewrite_pass.cc50 absl::Span<const DistributedTPURewriteHelpers::OutputDependency> in AddSynchronizationNode()
219 Graph* graph) -> Status { in Run()