- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for RemovedEdges (0.11 sec)
-
tensorflow/compiler/jit/shape_inference_helpers.h
// class can be used to remove cycles before running inference and replace them // after. Correct usage requires exactly one call to Remove(), followed by any // number of calls to RemovedEdges() and at most one call to Replace(). The call // to Replace() is optional if the graph will be discarded without being // executed, e.g., if it is being used purely for a shape inference pass. class BackEdgeHelper { public:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 12 18:06:51 UTC 2018 - 2.2K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference_helpers.cc
} } } } for (const BackEdge& be : back_edges_) { graph_->RemoveEdge(be.edge); } return absl::OkStatus(); } const std::vector<BackEdgeHelper::BackEdge>& BackEdgeHelper::RemovedEdges() const { return back_edges_; } Status BackEdgeHelper::Replace() { if (graph_ == nullptr) { return errors::Internal("BackEdgeHelper Replace called before Remove."); } if (replaced_) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 1.9K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference.cc
// the shape inference is complete. BackEdgeHelper back_edge; TF_RETURN_IF_ERROR(back_edge.Remove(graph)); TF_RETURN_IF_ERROR(PropagateShapes(graph, arg_shapes, back_edge.RemovedEdges(), &shape_refiner)); TF_RETURN_IF_ERROR(back_edge.Replace()); // Currently information does not flow "backward" from consumers to producers
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
TF_RETURN_IF_ERROR(back_edge_helper_.Remove(graph_.get())); VLOG(1) << "Found " << (back_edge_helper_.RemovedEdges().size()) << " backedges."; // Creates a map for quickly identifying whether a node output is a backedge. for (const auto& edge : back_edge_helper_.RemovedEdges()) { if (back_edge_node_output_.find(edge.src) != back_edge_node_output_.end() &&
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)