Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. RELEASE.md

    ## Known Caveats
    
    *   TF Core:
        *   A longstanding bug in `tf.while_loop`, which caused it to execute
            sequentially, even when `parallel_iterations>1`, has now been fixed.
            However, the increased parallelism may result in increased memory use.
            Users who experience unwanted regressions should reset their
            `while_loop`'s `parallel_iterations` value to 1, which is consistent
            with prior behavior.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 730.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

              while_op.getInput().getTypes(), while_op.getOutput().getTypes(),
              while_op.getBody().getArgumentTypes());
          return PropagateShapeToRegions(compatible_types,
                                         {&while_op.getCond(), &while_op.getBody()},
                                         max_iterations);
        }
        return PropagateShapeToRegions(while_op.getInput().getTypes(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

                    set_op.getItem());
                element_shape_acquired = true;
              } else if (TF::WhileOp while_op =
                             llvm::dyn_cast<TF::WhileOp>(use.getOwner())) {
                // Tensorlist is passed into a while loop, check inside the body
                // function.
                auto inside_uses = while_op.body_function()
                                       .getArgument(use.getOperandNumber())
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf.cc

      // Create the while op.
      auto while_op = builder->create<mhlo::WhileOp>(loc, init_types_with_loop_iv,
                                                     init_values_with_loop_iv);
      auto ivs_count = init_types_with_loop_iv.size();
    
      {
        OpBuilder::InsertionGuard guard(*builder);
    
        // Build up the only block in the condition region.
        Region &condition = while_op.getCond();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 291.8K bytes
    - Viewed (0)
Back to top