Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for next_itr (0.16 sec)

  1. 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)
  2. 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)
  3. 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)
Back to top