| /external/tensorflow/tensorflow/core/ir/ |
| D | utility_test.cc | 45 auto ret_op = cast<ReturnOp>(func.body().front().getTerminator()); in TEST()
80 auto ret_op = cast<ReturnOp>(func.body().front().getTerminator()); in TEST()
115 auto ret_op = cast<ReturnOp>(func.body().front().getTerminator()); in TEST()
|
| D | ops.cc | 444 if (!isa<ReturnOp>(getBody()->back())) in verifyBody()
449 ReturnOp return_op = cast<ReturnOp>(getBody()->getTerminator()); in verifyBody()
495 if (isa<ReturnOp>(op)) continue; in canonicalize()
780 LogicalResult ReturnOp::verify() { in verify()
781 ReturnOp op = *this; in verify()
792 ParseResult ReturnOp::parse(OpAsmParser &parser, OperationState &result) { in parse()
816 result.addAttribute(ReturnOp::control_ret_attrsAttrName(result.name), in parse()
828 void ReturnOp::print(OpAsmPrinter &printer) { in print()
852 void ReturnOp::build(OpBuilder &odsBuilder, OperationState &odsState, in build()
858 ReturnOp::control_ret_attrsAttrName(odsState.name), in build()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | cluster_outlining.cc | 48 void ReplaceClusterReturnWithReturn(tf_device::ReturnOp cluster_return_op, in ReplaceClusterReturnWithReturn()
50 builder->create<func::ReturnOp>(cluster_return_op.getLoc(), in ReplaceClusterReturnWithReturn()
93 cast<tf_device::ReturnOp>(outlined_func_block->getTerminator()); in BuildFunction()
|
| D | tfg-to-tfe.cc | 162 rewriter.create<func::ReturnOp>(loc); in matchAndRewrite()
230 rewriter.create<func::ReturnOp>( in matchAndRewrite()
240 class ConvertReturnOp : public OpConversionPattern<tfg::ReturnOp> {
244 tfg::ReturnOp ret, OpAdaptor adaptor, in matchAndRewrite()
506 target.addLegalOp<func::ReturnOp>(); in runOnOperation()
|
| D | xla_inline_device_ops.cc | 47 assert(mlir::isa<mlir::tf_device::ReturnOp>(return_op)); in InlineDeviceOp()
|
| D | xla_cluster_formation.cc | 45 builder.create<mlir::tf_device::ReturnOp>(call_op.getLoc(), in EncapsulatePartitionedCall()
|
| D | device_attribute_to_launch.cc | 49 builder.create<tf_device::ReturnOp>(op->getLoc(), op->getResults()) in WrapOpInLaunch()
|
| D | test_side_effect_analysis.cc | 69 if (llvm::isa<func::ReturnOp>(op)) { in MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID()
|
| D | outside_compiled_to_host_launch.cc | 51 .create<tf_device::ReturnOp>(host_op->getLoc(), host_op->getResults()) in WrapOpInLaunch()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| D | tf_to_tfrt_data.cc | 120 : public mlir::OpConversionPattern<mlir::func::ReturnOp> {
122 : OpConversionPattern<mlir::func::ReturnOp>(context) {} in ReturnOpConversion()
125 mlir::func::ReturnOp op, OpAdaptor adaptor, in matchAndRewrite()
127 rewriter.replaceOpWithNewOp<tfrt::compiler::ReturnOp>( in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/ |
| D | hlo_legalize_to_lhlo.cc | 289 auto returnOp = cast<mhlo::ReturnOp>(entryBlock.getTerminator()); in matchAndRewrite()
320 struct HloToLhloReturnOpConverter : public BaseOpConversion<mhlo::ReturnOp> {
322 using BaseOpConversion<mhlo::ReturnOp>::BaseOpConversion;
325 mhlo::ReturnOp op, OpAdaptor adaptor, in matchAndRewrite()
469 target.addDynamicallyLegalOp<mlir::func::ReturnOp>( in runOnOperation()
470 [&](mlir::func::ReturnOp op) { in runOnOperation()
|
| D | expand_hlo_tuples.cc | 100 auto returnOp = cast<mlir::func::ReturnOp>(func.getBody().back().back()); in expandTupledTensorInReturnOp()
122 builder.create<mlir::func::ReturnOp>(returnOp.getLoc(), in expandTupledTensorInReturnOp()
|
| /external/tensorflow/tensorflow/compiler/xla/mlir/transforms/runtime/ |
| D | convert_to_entrypoint.cc | 126 func.walk([&](func::ReturnOp ret) { in ConvertReturnOperations()
135 b.create<func::ReturnOp>(); in ConvertReturnOperations()
161 b.create<func::ReturnOp>(); in ConvertAssertOperations()
|
| D | custom_call_encoding.cc | 109 ib.create<LLVM::ReturnOp>(encoded); in PackString()
189 ib.create<LLVM::ReturnOp>(encoded); in PackDenseElementsAttribute()
209 ib.create<LLVM::ReturnOp>(data); in CreateGlobalFromArray()
242 ib.create<LLVM::ReturnOp>(encoded); in PackArrayAttribute()
300 ib.create<LLVM::ReturnOp>(data); in CreateGlobalFromDenseArray()
334 ib.create<LLVM::ReturnOp>(encoded); in PackDenseArrayAttribute()
362 ib.create<LLVM::ReturnOp>(encoded); in PackEmptyArrayAttribute()
790 b.create<LLVM::ReturnOp>(arr); in EncodeAttributes()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | target_annotation.cc | 114 !llvm::isa<func::ReturnOp, func::FuncOp, CallableOpInterface>(op)) { in SetTargetAnnotation()
132 !llvm::isa<func::ReturnOp, func::FuncOp, CallOpInterface>(op)) { in runOnFunction()
|
| D | get_alternative_subgraph.cc | 205 !llvm::isa<func::ReturnOp, func::FuncOp, CallOpInterface>(op) && in IsAllSupportedbySpec()
259 !llvm::isa<func::ReturnOp, func::FuncOp, CallableOpInterface>(op)) { in GetAlternativeViewForSpec()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils_test.cc | 155 mlir::func::ReturnOp::getOperationName()); in TEST_F()
209 mlir::func::ReturnOp::getOperationName()); in TEST_F()
236 mlir::func::ReturnOp::getOperationName()); in TEST_F()
|
| D | tftext_utils.cc | 140 builder.create<func::ReturnOp>(func.getLoc(), op.getResults()); in ConvertWhitespaceTokenizer()
269 builder.create<func::ReturnOp>(func.getLoc(), op.getResults()); in ConvertNgrams()
350 builder.create<func::ReturnOp>(func.getLoc(), op.getResults()); in ConvertSgnnProjection()
|
| D | perception_ops_utils.cc | 106 builder.create<func::ReturnOp>(func_.getLoc(), op.getResults()); in RewriteFunc()
202 builder.create<func::ReturnOp>(func_.getLoc(), op.getResults()); in RewriteFunc()
|
| D | nms_utils.cc | 57 builder.create<mlir::func::ReturnOp>(func_.getLoc(), op.getResults()); in RewriteFunc()
102 builder.create<func::ReturnOp>(func_.getLoc(), op.getResults()); in RewriteFunc()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_control_flow.cc | 125 builder.create<mhlo::ReturnOp>(loc, result.getResults()); in ImportXlaRegion()
128 builder.create<mhlo::ReturnOp>(loc, tuple_op.getResult()); in ImportXlaRegion()
300 builder->create<mhlo::ReturnOp>(loc, tuple_results.getResult()); in ReplaceTerminator()
302 builder->create<mhlo::ReturnOp>(loc, results); in ReplaceTerminator()
|
| D | utils.h | 44 builder->create<ReturnOp>(loc, reducer.getResult()); in BuildReduceBody()
|
| D | xla_legalize_tf_no_fallback.cc | 74 target.addLegalOp<ModuleOp, func::FuncOp, ::mlir::func::ReturnOp>(); in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/ |
| D | execution_metadata_exporter.cc | 63 if (llvm::isa<mlir::func::ReturnOp, mlir::quantfork::StatisticsOp>(op)) in GetDeviceName()
106 if (llvm::isa<mlir::func::ReturnOp, mlir::quantfork::StatisticsOp>(&inst)) in CreateSubgraphMetadata()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_device_ops.td | 55 [SingleBlockImplicitTerminator<"ReturnOp">]> {
133 [SingleBlockImplicitTerminator<"ReturnOp">]> {
181 [SingleBlockImplicitTerminator<"ReturnOp">, AttrSizedOperandSegments]> {
313 [SingleBlockImplicitTerminator<"ReturnOp">]> {
375 [SingleBlockImplicitTerminator<"ReturnOp">]> {
|