| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | constant_fold.cc | 63 int64_t results_size = get_size(inst->getResultTypes()); in ShouldBeFolded()
83 for (auto& type : inst->getResultTypes()) { in ConstantFoldFallbackHook()
97 llvm::all_of(inst->getResultTypes(), [](Type ty) { in ConstantFoldFallbackHook()
104 for (Type ty : inst->getResultTypes()) { in ConstantFoldFallbackHook()
|
| D | functional_control_flow_to_regions.cc | 100 if_op.getLoc(), if_op.getResultTypes(), cond, if_op.is_stateless(), in ConvertIfOp()
118 while_op.getLoc(), while_op.getResultTypes(), while_op.input(), in ConvertWhileOp()
|
| D | fold_broadcast.cc | 137 if (op->getNumOperands() != 2 || op->getResultTypes().size() != 1) in RewriteOp()
142 op->getResultTypes().front().dyn_cast_or_null<RankedTensorType>(); in RewriteOp()
|
| D | tpu_merge_variables_with_execute.cc | 348 output_types.append(merged_execute_launch.getResultTypes().begin(), in ReplaceParallelExecute()
349 merged_execute_launch.getResultTypes().end()); in ReplaceParallelExecute()
510 for (const auto& type : execute_launch.getResultTypes()) { in MergeForOneTPUExecute()
554 merged_execute.getResultTypes()); in MergeForOneTPUExecute()
|
| D | outside_compiled_to_host_launch.cc | 66 /*result_types=*/host_op->getResultTypes()); in WrapOpInLaunch()
|
| /external/llvm-project/mlir/lib/Dialect/SCF/Transforms/ |
| D | StructuralTypeConversions.cpp | 27 for (auto type : op.getResultTypes()) { in matchAndRewrite()
95 for (auto type : op.getResultTypes()) { in matchAndRewrite()
142 return typeConverter.isLegal(op->getResultTypes()); in populateSCFStructuralTypeConversionsAndLegality()
|
| D | Utils.cpp | 99 FunctionType::get(values.getTypes(), ifOp.getResultTypes(), ctx); in outlineIfOp()
112 b.create<ReturnOp>(loc, term->getResultTypes(), terminatorOperands); in outlineIfOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf_control_flow.cc | 108 auto result_type = builder.getTupleType(op.getResultTypes()); in LowerIf()
137 builder.create<mhlo::CaseOp>(loc, op.getResultTypes(), op.branch_index(), in LowerCase()
163 loc, builder.getTupleType(op.getResultTypes()), tuple_input); in LowerWhile()
272 auto result_type = builder.getTupleType(op.getResultTypes()); in LowerIfRegion()
302 builder.create<mhlo::CaseOp>(loc, op.getResultTypes(), op.branch_index(), in LowerCaseRegion()
328 auto while_result_types = llvm::to_vector<4>(op.getResultTypes()); in LowerWhileRegion()
|
| D | legalize_tf_types.cc | 107 llvm::any_of(op->getResultTypes(), isIllegalType)) in isDynamicallyLegal()
138 if (failed(getTypeConverter()->convertTypes(op->getResultTypes(), in matchAndRewrite()
|
| D | prepare_for_export.cc | 56 ShapedType return_type = op->getResultTypes().front().cast<ShapedType>(); in runOnFunction()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | TypeRange.cpp | 24 : TypeRange(values.getBase()->getResultTypes().slice(values.getStartIndex(), in TypeRange()
31 this->base = op->getResultTypes().drop_front(owner.startIndex).data(); in TypeRange()
|
| D | Operation.cpp | 560 auto Operation::getResultTypes() -> result_type_range { in getResultTypes() function in Operation
622 auto *newOp = create(getLoc(), getName(), getResultTypes(), operands, attrs, in cloneWithoutRegions()
877 for (auto resultType : op->getResultTypes()) { in verifySameOperandsAndResultShape()
935 for (auto resultType : op->getResultTypes().drop_front(1)) { in verifySameOperandsAndResultType()
1003 for (auto resultType : op->getResultTypes()) { in verifyResultsAreBoolLike()
1014 for (auto resultType : op->getResultTypes()) in verifyResultsAreFloatLike()
1023 for (auto resultType : op->getResultTypes()) in verifyResultsAreSignlessIntegerLike()
1100 llvm::make_filter_range(op->getResultTypes(), isMappableType)); in verifyElementwiseMappable()
|
| D | OperationSupport.cpp | 509 return getBase()->getResultTypes().slice(getStartIndex(), size()); in getTypes()
561 ArrayRef<Type> resultTypes = op->getResultTypes(); in computeHash()
603 ArrayRef<Type> lhsResultTypes = lhs->getResultTypes(); in isEquivalentTo()
604 ArrayRef<Type> rhsResultTypes = rhs->getResultTypes(); in isEquivalentTo()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | ElementwiseToLinalg.cpp | 45 op, /*resultTensorTypes=*/op->getResultTypes(), in matchAndRewrite()
57 llvm::map_range(op->getResultTypes(), [](Type type) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_traits.h | 54 for (auto result_type : llvm::drop_begin(op->getResultTypes(), 1)) { in verifyTrait()
84 for (const auto& result_type : op->getResultTypes()) { in verifySameOperandsAndResultElementTypeResolveRef()
|
| /external/llvm-project/mlir/test/lib/IR/ |
| D | TestInterfaces.cpp | 22 for (Type type : op->getResultTypes()) { in runOnOperation()
|
| /external/llvm-project/mlir/lib/Dialect/ |
| D | Traits.cpp | 193 auto resultsHasTensorVectorType = hasTensorOrVectorType(op->getResultTypes()); in verifyCompatibleOperandBroadcast()
220 op->getResultTypes(), [](Type t) { return t.isa<RankedTensorType>(); }); in verifyCompatibleOperandBroadcast()
|
| /external/llvm-project/mlir/examples/toy/Ch5/mlir/ |
| D | ShapeInferencePass.cpp | 103 return llvm::any_of(op->getResultTypes(), [](Type resultType) { in returnsDynamicShape()
|
| /external/llvm-project/mlir/examples/toy/Ch4/mlir/ |
| D | ShapeInferencePass.cpp | 103 return llvm::any_of(op->getResultTypes(), [](Type resultType) { in returnsDynamicShape()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | raise_custom_ops.cc | 56 op->getLoc(), op->getResultTypes(), op->getOperands()); in runOnFunction()
|
| /external/llvm-project/mlir/examples/toy/Ch7/mlir/ |
| D | ShapeInferencePass.cpp | 103 return llvm::any_of(op->getResultTypes(), [](Type resultType) { in returnsDynamicShape()
|
| /external/llvm-project/mlir/examples/toy/Ch6/mlir/ |
| D | ShapeInferencePass.cpp | 103 return llvm::any_of(op->getResultTypes(), [](Type resultType) { in returnsDynamicShape()
|
| /external/llvm-project/mlir/lib/Interfaces/ |
| D | InferTypeOpInterface.cpp | 57 op->getResultTypes())) in verifyInferredResultTypes()
|
| /external/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
| D | AsyncRegionRewriter.cpp | 90 copy(op->getResultTypes(), std::back_inserter(resultTypes)); in rewriteAsyncOp()
168 copy(executeOp.getResultTypes(), std::back_inserter(resultTypes)); in addAsyncTokenResults()
|
| /external/llvm-project/mlir/lib/Dialect/Shape/Transforms/ |
| D | StructuralTypeConversions.cpp | 65 return typeConverter.isLegal(op.getResultTypes()); in populateShapeStructuralTypeConversionsAndLegality()
|