- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 36 for while_loop (0.27 sec)
-
tensorflow/compiler/jit/shape_inference_test.cc
{"D", {PartialTensorShape({2, 3})}}, }; TF_EXPECT_OK(ShapeAnnotationsMatch(*graph, shape_info, expected)); } TEST(ShapeInferenceTest, WhileLoop) { // Graph: // x = array_ops.placeholder(dtypes.int32) // y = control_flow_ops.while_loop(lambda i: i < 10, lambda i: i + 1, [x]) Graph graph(OpRegistry::Global()); { Scope scope = Scope::NewRootScope().ExitOnError();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 00:41:19 UTC 2024 - 10.3K bytes - Viewed (0) -
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) -
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) -
tensorflow/c/c_api.cc
#include "tensorflow/c/experimental/filesystem/modular_filesystem.h" #include "tensorflow/cc/framework/gradients.h" #include "tensorflow/cc/framework/ops.h" #include "tensorflow/cc/framework/scope_internal.h" #include "tensorflow/cc/ops/while_loop.h" #include "tensorflow/cc/saved_model/loader.h" #include "tensorflow/core/distributed_runtime/server_lib.h" #include "tensorflow/core/framework/logging.h" #include "tensorflow/core/framework/op_gen_lib.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/while_loop_outline.cc
for (auto extra_operand : extra_operands) new_types.push_back(extra_operand.getType()); auto new_while_op = OpBuilder(while_op).create<WhileOp>( while_op.getLoc(), new_types, operands, while_op->getAttrs()); new_while_op.getCond().takeBody(while_op.getCond()); new_while_op.getBody().takeBody(while_op.getBody()); while_op.replaceAllUsesWith(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
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) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
for (int64_t i = 0; i < while_op.getNumResults(); ++i) { auto it = buffer_to_size->find(while_op.getOperand(i)); if (it == buffer_to_size->end()) continue; new_while_operands.push_back(it->getSecond().size); } auto new_while = builder.create<TF::WhileRegionOp>( while_op.getLoc(), body_region.front().getTerminator()->getOperandTypes(), new_while_operands, while_op->getAttrs());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/functional_control_flow_to_regions.cc
} LogicalResult ConvertWhileOp(WhileOp while_op, bool allow_passthrough_args) { auto while_region = OpBuilder(while_op).create<TF::WhileRegionOp>( while_op.getLoc(), while_op.getResultTypes(), while_op.getInput(), while_op.getParallelIterations(), while_op.getIsStateless(), while_op.getShapeInvariant()); CopyDeviceAndUnderscoredAttributes(while_op, while_region); YieldOp cond_yield =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc
id_map[i] = j++; new_operands.push_back(while_op.getOperand(i)); new_result_types.push_back(while_op.getResultTypes()[i]); } } auto new_while_op = OpBuilder(while_op).create<WhileOp>( while_op.getLoc(), new_result_types, new_operands, while_op->getAttrs()); new_while_op.getCond().takeBody(while_op.getCond()); new_while_op.getBody().takeBody(while_op.getBody()); for (auto i = 0; i < n; ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
read_var->erase(); } else if (auto while_op = dyn_cast<TF::WhileOp>(user_op)) { GetUpdatedWhileOp<TF::WhileOp>( while_op, while_op.cond_function().getArgumentTypes(), args_to_erase); while_op->erase(); } else if (auto while_op = dyn_cast<TF::WhileRegionOp>(user_op)) { auto new_while_op = GetUpdatedWhileOp( while_op, while_op.getCond().getArgumentTypes(), args_to_erase);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0)