- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 60 for while_loop (0.3 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/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) -
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/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) -
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) -
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) -
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)