- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for handleTerminator (0.27 sec)
-
tensorflow/compiler/mlir/tfrt/ir/mlrt/tf_mlrt_ops.cc
mlir::IRMapping &mapping) const final { // All tf_mlrt dialect ops can be inlined. return true; } // Note that CallOp and ReturnOp are handled by func; so need to implement // handleTerminator. }; } // namespace TensorflowMlrtDialect::TensorflowMlrtDialect(mlir::MLIRContext *context) : mlir::Dialect(/*name=*/"tf_mlrt", context, mlir::TypeID::get<TensorflowMlrtDialect>()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/ir/tfr_ops.cc
return true; } // Handle the given inlined terminator by replacing it with a new operation // as necessary. Required when the region has only one block. void handleTerminator(Operation *op, ValueRange valuesToRepl) const final { auto retValOp = dyn_cast<TFRReturnOp>(op); if (!retValOp) return; for (auto ret_value : llvm::zip(valuesToRepl, retValOp.getOperands())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Nov 21 16:55:41 UTC 2023 - 38.2K bytes - Viewed (0)