- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 60 for while_loop (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/convert_control_to_data_outputs.cc
auto loc = NameLoc::get( builder.getStringAttr("chain_control_outputs@" + while_op.getBody())); IslandOp const_wrapper = GetDummyConstant(builder, const_type, loc); // Get new operand and result types. auto new_operands = llvm::to_vector<4>(while_op->getOperands()); auto new_result_types = llvm::to_vector<4>(while_op->getResultTypes()); Value const_output = const_wrapper.getOutputs()[0];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 28.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_device_inference.cc
// Propagation to callees. auto walk_res = func_op.walk([&](Operation* op) { if (auto while_op = dyn_cast<WhileOp>(op)) { if (failed(propagate_operands_to_callee_arguments( while_op, while_op.getOperands(), {while_op.body_function(), while_op.cond_function()}, func_res))) return WalkResult::interrupt();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 03:47:00 UTC 2023 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc
NameAttrList cond_fn, body_fn; cond_fn.set_name("f3"); body_fn.set_name("f2"); auto while_op = ops::While(s.WithOpName("while"), std::initializer_list<Input>{arg0, arg1}, cond_fn, body_fn); auto ret2 = ops::_Retval(s.WithOpName("ret2"), while_op.output[0], 2); auto ret3 = ops::_Retval(s.WithOpName("ret3"), while_op.output[1], 3); std::unique_ptr<Graph> g(new Graph(OpRegistry::Global()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/analysis/resource_alias_analysis.cc
PropagateInputToOutput(op->getOperand(result.getResultNumber()), result); } else if (auto while_op = dyn_cast<WhileOp>(op)) { AnalyzeWhileLoop(while_op, backtrack_analysis.GetAnalysisForFunc( while_op.body_function())); } else if (auto while_region = dyn_cast<WhileRegionOp>(op)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 15 09:04:13 UTC 2024 - 28.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
module.walk([&](mlir::TFL::WhileOp while_op) { auto cond = symbol_table.lookup<mlir::func::FuncOp>( while_op->getAttr("cond").cast<mlir::FlatSymbolRefAttr>().getValue()); AddCallOpInWhileOpRegion(while_op.getCond(), cond); while_op->removeAttr("cond"); auto body = symbol_table.lookup<mlir::func::FuncOp>( while_op->getAttr("body").cast<mlir::FlatSymbolRefAttr>().getValue());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
// Recreate the while op. OpBuilder builder(while_op); // Now use the filtered original operands, which will be replaced by // AddLoadsStoresOutsideControlFlowOp(). auto new_while = builder.create<TF::WhileOp>( while_op.getLoc(), body.getFunctionType().getResults(), FilterRange<Value, OperandRange>(while_op.getOperands(), resource_arg_uses),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
MarkOutsideCompiled(host_if.getOperation()); } if (auto while_op = llvm::dyn_cast<mlir::TF::WhileRegionOp>(op)) { if (!HasOutsideCompilationNested(op)) return WalkResult::advance(); OpBuilder builder(while_op); auto host_while = CloneEmptyWhile(while_op.getParallelIterations(), while_op.getLoc(), builder); const auto condition_send_recv_key =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
TF::WhileOp* while_op) { // Given a while loop body function 'body_func', find the tf.While Op that // uses it. auto uses_optional = body_func.getSymbolUses(module); if (!uses_optional.has_value()) { body_func.emitOpError() << "no use of while loop body"; return LogicalResult::failure(); } *while_op = nullptr; for (auto& use : uses_optional.value()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
LogicalResult matchAndRewrite(WhileRegionOp while_op, PatternRewriter &rewriter) const override { auto &body_block = while_op.getBody().front(); auto &cond_block = while_op.getCond().front(); bool changed = false; for (int op_idx : llvm::seq<int>(0, while_op.getNumOperands())) { auto body_arg = body_block.getArgument(op_idx);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_weights.cc
if (!func) continue; next_values_to_visit.push_back( func.getArgument(next_op_operand_num)); } else if (auto while_op = llvm::dyn_cast_or_null<TF::WhileOp>(next_op)) { func::FuncOp func = while_op.body_function(); auto func_argument = func.getArgument(next_op_operand_num); // Check if the op is returned without mutation. Returning values
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 07:39:40 UTC 2024 - 11.3K bytes - Viewed (0)