- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for getElseBranch (0.39 sec)
-
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
LogicalResult IfRegionOp::verifyRegions() { IfRegionOp op = *this; TypeRange then_types = op.getThenBranch().front().getTerminator()->getOperandTypes(); TypeRange else_types = op.getElseBranch().front().getTerminator()->getOperandTypes(); TypeRangeWithDesc results{op.getResultTypes(), "result"}; TypeRangeWithDesc then_results{then_types, "then result"};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
/*_else_func_name=*/nullptr); new_op.getThenBranch().takeBody(op.getTrueBranch()); new_op.getElseBranch().takeBody(op.getFalseBranch()); ReplaceReturnOp(new_op.getThenBranch(), rewriter); ReplaceReturnOp(new_op.getElseBranch(), rewriter); rewriter.replaceOp(op, new_op.getResults()); return success(); } }; // Converts mhlo.pad to tf.PadV2
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
auto opcode_index = GetOpcodeIndex("if", tflite::BuiltinOperator_IF); int then_subgraph_index = subgraph_index_map_.at(op.getThenBranch().str()); int else_subgraph_index = subgraph_index_map_.at(op.getElseBranch().str()); auto builtin_options = tflite::CreateIfOptions(builder_, then_subgraph_index, else_subgraph_index) .Union();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
} bool ShapeInference::InferShapeForIfRegion(IfRegionOp op) { bool changed = false; Operation* then_yield = op.getThenBranch().front().getTerminator(); Operation* else_yield = op.getElseBranch().front().getTerminator(); for (auto result : zip(op.getResults(), then_yield->getOperandTypes(), else_yield->getOperandTypes())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)