- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 62 for while_loop (0.14 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
llvm::SmallVector<std::pair<int64_t, llvm::SmallVector<Value, 4>>, 4> AnnotateCompileOpAndGetExecuteArgToWhileArgsMapping( TF::WhileRegionOp while_op, tf_device::ReplicateOp replicate, TF::TPUExecuteAndUpdateVariablesOp execute, tf_device::LaunchOp compile_launch) { Region& body = while_op.getBody(); Region& cond = while_op.getCond(); llvm::SmallVector<std::pair<int64_t, llvm::SmallVector<Value, 4>>, 4> mapping;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_array_ops_decomposition.cc
for (int64_t i = 0; i < while_op.getNumOperands(); ++i) { if (ta_arg_buffer_type(i)) { while_op.getResult(i).replaceAllUsesWith(while_op.getOperand(i)); } else { while_op.getResult(i).replaceAllUsesWith(new_while.getResult(i)); } } while_op.erase(); return success(); } LogicalResult HandleIfOp(TF::IfOp if_op, ModuleOp module,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0) -
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/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/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/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)