- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 42 for while_v1 (7.86 sec)
-
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/vhlo.mlir
//CHECK-NEXT: return %0#0 : tensor<448xf32> //CHECK-NEXT:} func.func @while(%init_i: tensor<i64>, %init_sum: tensor<i64>) -> tensor<i64>{ %0, %1 = "vhlo.while_v1"(%init_i, %init_sum) ({ ^bb0(%arg0: tensor<i64>, %arg1: tensor<i64>): %cond = "vhlo.compare_v1"(%arg0, %arg1) { comparison_direction = #vhlo<comparison_direction_v1 LT>,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 31.9K bytes - Viewed (1) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
}); module.walk([&](mlir::vhlo::WhileOpV1 while_op) { auto cond = symbol_table.lookup<mlir::func::FuncOp>( while_op->getAttr("cond").cast<mlir::FlatSymbolRefAttr>().getValue()); InlineVhloOpRegion(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/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/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/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)