- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 60 for while_loop (0.15 sec)
-
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) -
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)