- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for BackEdge (0.09 sec)
-
tensorflow/compiler/jit/shape_inference_helpers.cc
if (e->src()->IsNextIteration()) { back_edges_.push_back( BackEdge{e, e->src(), e->src_output(), e->dst(), e->dst_input()}); } } } } for (const BackEdge& be : back_edges_) { graph_->RemoveEdge(be.edge); } return absl::OkStatus(); } const std::vector<BackEdgeHelper::BackEdge>& BackEdgeHelper::RemovedEdges() const { return back_edges_; }
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_helpers.h
Status Remove(Graph* graph); // Gets the list of removed edges. const std::vector<BackEdge>& RemovedEdges() const; // Replaces the back edges removed by a prior call to Remove. Status Replace(); private: Graph* graph_ = nullptr; // not owned std::vector<BackEdge> back_edges_; // Set once Replace has been called. bool replaced_ = false; }; } // namespace tensorflow
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/mlir/tensorflow/tests/graphdef2mlir/graph-while-loop.pbtxt
# RUN: tf-mlir-translate -graphdef-to-mlir -tf-enable-shape-inference-on-import=false -mlir-print-debuginfo %s -o - | FileCheck %s # Verify that importing a Graph with a backedge leads to two NextIteration nodes # to break the cycle. # CHECK-LABEL: func @main() # CHECK: %[[NEXTITERATION:[a-z0-9]+]], %[[NEXTITERATION_token:[a-z0-9]+]], {{.*}} = tf_executor.NextIteration.Source # CHECK: tf_executor.Merge {{.*}} %[[NEXTITERATION]]
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 27 18:14:13 UTC 2020 - 3K bytes - Viewed (0) -
tensorflow/cc/ops/while_loop.cc
result[i] = outputs[i].node(); } return result; } // Manually generates the name of the `loop_var_idx`-th NextIteration node of a // loop being constructed with `scope`. This is used to define the backedge // before the NextIteration node is created. string NextIterationName(const Scope& scope, int loop_var_idx) { string result; const string& prefix = scope.impl()->name();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 01:01:21 UTC 2024 - 9.5K bytes - Viewed (0)