- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 62 for while_loop (0.23 sec)
-
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) -
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/transforms/functional_control_flow_to_regions.cc
} LogicalResult ConvertWhileOp(WhileOp while_op, bool allow_passthrough_args) { auto while_region = OpBuilder(while_op).create<TF::WhileRegionOp>( while_op.getLoc(), while_op.getResultTypes(), while_op.getInput(), while_op.getParallelIterations(), while_op.getIsStateless(), while_op.getShapeInvariant()); CopyDeviceAndUnderscoredAttributes(while_op, while_region); YieldOp cond_yield =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/remove_unused_while_results.cc
}; // Prunes result defining op if possible, returns true if pruning was done. bool TryPruneResultDefiningOp(TF::WhileRegionOp while_op, OpResult result) { // Don't prune if result is used. if (!result.use_empty()) return false; Block& body_block = while_op.getBody().front(); Block& cond_block = while_op.getCond().front(); Operation* body_yield_op = body_block.getTerminator();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Nov 16 01:49:07 UTC 2022 - 5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_while_operands.cc
id_map[i] = j++; new_operands.push_back(while_op.getOperand(i)); new_result_types.push_back(while_op.getResultTypes()[i]); } } auto new_while_op = OpBuilder(while_op).create<WhileOp>( while_op.getLoc(), new_result_types, new_operands, while_op->getAttrs()); new_while_op.getCond().takeBody(while_op.getCond()); new_while_op.getBody().takeBody(while_op.getBody()); for (auto i = 0; i < n; ++i) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf_while.cc
func.setPrivate(); } void RunOnWhile(TF::WhileOp while_op) { Operation* op = while_op.getOperation(); // Create new TFL While op that will be used to replace TF While op. auto new_op = OpBuilder(op).create<TFL::WhileOp>( op->getLoc(), op->getResultTypes(), op->getOperands(), while_op.getIsStateless()); Location loc = while_op->getLoc(); CreateRegionWithCall(while_op.cond_function(), new_op.getCond(), loc);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Nov 03 12:35:38 UTC 2022 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_freeze_variables.cc
read_var->erase(); } else if (auto while_op = dyn_cast<TF::WhileOp>(user_op)) { GetUpdatedWhileOp<TF::WhileOp>( while_op, while_op.cond_function().getArgumentTypes(), args_to_erase); while_op->erase(); } else if (auto while_op = dyn_cast<TF::WhileRegionOp>(user_op)) { auto new_while_op = GetUpdatedWhileOp( while_op, while_op.getCond().getArgumentTypes(), args_to_erase);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 09:56:53 UTC 2024 - 19.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
// Create the new while op. auto new_while_operands = llvm::to_vector<8>(while_op.getOperands()); OpBuilder builder(while_op); assert(while_op.getNumOperands() == while_op.getNumResults()); for (int64_t i = 0; i < while_op.getNumResults(); ++i) { auto it = data_var_to_size_var.find(while_op.getOperand(i)); if (it == data_var_to_size_var.end()) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0)