- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for while_loop (0.24 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/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/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/analysis/tf_dataflow.h
while_op, while_op.getCondAttr()); func::FuncOp body = SymbolTable::lookupNearestSymbolFrom<func::FuncOp>( while_op, while_op.getBodyAttr()); for (auto &arg : while_op->getOpOperands()) { BlockArgument cond_arg = cond.getArgument(arg.getOperandNumber()); this->join(getLatticeElement(cond_arg), *getLatticeElement(arg.get()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 06 23:53:00 UTC 2024 - 3.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.h
// Analyzes while loops to compute resource IDs for the loop results. // `body_info` is the backtrack analysis info for the loop body. void AnalyzeWhileLoop(Operation* while_op, const BacktrackAnalysisInfo& body_info); // Analyzes tf.Case/tf.If ops to compute resource IDs. template <class CaseOrIfOp> void AnalyzeFunctionalCaseOrIfOp(CaseOrIfOp case_or_if_op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 7.6K bytes - Viewed (0)