- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for next_itr (0.13 sec)
-
tensorflow/compiler/jit/xla_cluster_util_test.cc
ops::internal::Enter enter(s.WithOpName("init_value"), init_value, loop_name); ops::Merge merge(s.WithOpName("merge"), {init_value, init_value}); Output next_iteration = ops::NextIteration(s.WithOpName("next_itr"), merge.output); return s.graph()->UpdateEdge(next_iteration.node(), 0, merge.output.node(), 1); } TEST(NodesRelatedToRefVariables, Cycles) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 09:53:30 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/compiler/jit/shape_inference.cc
// directly. auto iter = merge_to_next_iteration.find(n); if (iter != merge_to_next_iteration.end()) { const Node *next_iter = iter->second, *node = next_iter; do { TF_RETURN_IF_ERROR(node->input_node(0, &node)); } while (node->IsIdentity()); const Node* switch_input; bool is_loop_invariant = node->IsSwitch() &&
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/ir/tf_traits.h
llvm::SmallDenseMap<mlir::TF::ResourceHandle, int64_t>& resource_handle_id_map, int64_t& next_id) { llvm::SmallVector<mlir::TF::ResourceHandleValueAndId> resource_vec; for (Value resource : mlir::tf_type::filter_resources(this->getOperation()->getResults())) { resource_vec.push_back({resource, next_id++}); } return resource_vec; } }; } // namespace TF
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0)