Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 14 for while_loop (0.2 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/cc/ops/while_loop.cc 

    See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/

#include "tensorflow/cc/ops/while_loop.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/core/common_runtime/shape_refiner.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Feb 26 01:01:21 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  3. github.com/tensorflow/tensorflow

    tensorflow/c/experimental/saved_model/internal/testdata/gen_saved_models.py 

      class Module(module.Module):
    """A module with a while loop."""

    @def_function.function(
        input_signature=[tensor_spec.TensorSpec((), dtypes.float32)])
    def compute(self, value):
      acc, _ = while_loop.while_loop(
          cond=lambda acc, i: i > 0,
          body=lambda acc, i: (acc + i, i - 1),
          loop_vars=(constant_op.constant(0.0), value))
      return acc

  to_save = Module()
  saved_model.save(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 06 21:32:57 UTC 2023
    - 3.4K bytes
    - Viewed (0)
  4. github.com/tensorflow/tensorflow

    tensorflow/c/python_api.h 

    // This method is used to add a new input edge to 'dst', which must be a While
// op. The While op's "T" attribute must have already been updated to include
// the new edge. This is used to construct tf.while_loop gradients.
void AddWhileInputHack(TF_Graph* graph, TF_Output new_src, TF_Operation* dst,
                       TF_Status* status);

}  // namespace tensorflow
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jul 12 18:48:56 UTC 2023
    - 3.5K bytes
    - Viewed (0)
  5. github.com/tensorflow/tensorflow

    tensorflow/cc/ops/while_loop_test.cc 

    See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/

#include "tensorflow/cc/ops/while_loop.h"
#include "tensorflow/cc/client/client_session.h"
#include "tensorflow/cc/ops/standard_ops.h"
#include "tensorflow/core/framework/tensor_testutil.h"
#include "tensorflow/core/graph/while_context.h"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 13 22:30:58 UTC 2023
    - 6.4K bytes
    - Viewed (0)
  6. github.com/tensorflow/tensorflow

    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)
  7. github.com/JetBrains/kotlin

    analysis/analysis-api/testData/components/expressionInfoProvider/isUsedAsExpression/whileLoop.kt 

    Kristoffer Andersen <******@****.***> 1659368814 +0200
    Registered: Wed Jun 12 09:53:16 UTC 2024
    - Last Modified: Wed Oct 05 15:06:52 UTC 2022
    - 81 bytes
    - Viewed (0)
  8. github.com/JetBrains/kotlin

    analysis/analysis-api/testData/components/expressionInfoProvider/isUsedAsExpression/whileLoop.txt 

    Kristoffer Andersen <******@****.***> 1659368814 +0200
    Registered: Wed Jun 12 09:53:16 UTC 2024
    - Last Modified: Wed Oct 05 15:06:52 UTC 2022
    - 86 bytes
    - Viewed (0)
  9. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/lite/tests/mlir2flatbuffer/while_op.mlir 

    Meghna Natraj <******@****.***> 1657816346 -0700
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 14 16:41:28 UTC 2022
    - 7.5K bytes
    - Viewed (0)
  10. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/while_op.mlir 

    Meghna Natraj <******@****.***> 1657816346 -0700
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jul 14 16:41:28 UTC 2022
    - 2K bytes
    - Viewed (0)
  11. github.com/tensorflow/tensorflow

    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)
Back to top