Searched refs:op_handler_name (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | corert_converter.cc | 142 result.op_handler_name = tf_device_name; in ParseDeviceName()
161 mlir::Operation *op, llvm::StringRef op_handler_name, in ConvertOpHandler() argument
163 auto iter = op_handler_by_name_.find(op_handler_name); in ConvertOpHandler()
174 op_handler_name); in ConvertOpHandler()
175 op_handler_by_name_[op_handler_name] = get_op_handler_op.getResult(); in ConvertOpHandler()
|
| D | corert_converter.h | 35 std::string op_handler_name; member
|
| D | tf_to_tfrt.cc | 177 op, operands, parsed_device_name->op_handler_name, op_attrs, in matchAndRewrite()
225 mlir::Operation *op, ValueRange operands, llvm::StringRef op_handler_name,
306 mlir::Operation *op, ValueRange operands, llvm::StringRef op_handler_name, in ConvertToCoreRTExecuteOp() argument
322 corert_converter_.ConvertOpHandler(op, op_handler_name, &rewriter); in ConvertToCoreRTExecuteOp()
860 op, parsed_device_name->op_handler_name, &rewriter); in matchAndRewrite()
|