Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for while_loop (0.12 sec)

  1. tensorflow/cc/framework/while_gradients.cc

    #include "tensorflow/cc/framework/gradients.h"
    #include "tensorflow/cc/framework/scope_internal.h"
    #include "tensorflow/cc/ops/control_flow_ops_internal.h"
    #include "tensorflow/cc/ops/standard_ops.h"
    #include "tensorflow/cc/ops/while_loop.h"
    
    namespace tensorflow {
    namespace {
    
    using ops::BodyGraphBuilderFn;
    using ops::BuildWhileLoop;
    using ops::CondGraphBuilderFn;
    
    Output ToOutput(OutputTensor output_tensor) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 13 05:57:22 UTC 2024
    - 8.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/analysis/resource_value_typed_analyzer.cc

          return;
        }
        if (auto while_op = dyn_cast<TF::WhileOp>(op)) {
          for (auto callee : {while_op.cond_function(), while_op.body_function()}) {
            PropagatePotentiallyWrittenUpFromCallee(callee.getRegion(),
                                                    while_op.getInput());
          }
          return;
        }
        if (auto while_op = dyn_cast<TF::WhileRegionOp>(op)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 09:04:13 UTC 2024
    - 8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.h

      // Analyzes while loops to compute resource IDs for the loop results.
      // `body_info` is the backtrack analysis info for the loop body.
      void AnalyzeWhileLoop(Operation* while_op,
                            const BacktrackAnalysisInfo& body_info);
    
      // Analyzes tf.Case/tf.If ops to compute resource IDs.
      template <class CaseOrIfOp>
      void AnalyzeFunctionalCaseOrIfOp(CaseOrIfOp case_or_if_op,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 15 09:04:13 UTC 2024
    - 7.6K bytes
    - Viewed (0)
Back to top