- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 42 for while_loop (0.28 sec)
-
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) -
tensorflow/cc/BUILD
"//tensorflow/core:test", "//tensorflow/core:test_main", "//tensorflow/core:testlib", ], ) cc_library_with_android_deps( name = "while_loop", srcs = ["ops/while_loop.cc"], hdrs = ["ops/while_loop.h"], android_deps = [ "//tensorflow/core:portable_tensorflow_lib", ], common_deps = [ ":cc_ops", ":cc_ops_internal", ":ops",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 27 18:00:18 UTC 2024 - 23.5K bytes - Viewed (0) -
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) -
tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test_base.py
from tensorflow.python.ops import nn_ops from tensorflow.python.ops import random_ops from tensorflow.python.ops import string_ops from tensorflow.python.ops import variables from tensorflow.python.ops import while_loop as while_loop_ops from tensorflow.python.ops.ragged import ragged_string_ops from tensorflow.python.platform import test from tensorflow.python.platform import tf_logging as logging
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 21 08:51:46 UTC 2024 - 51.2K 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)