Home
last modified time | relevance | path

Searched refs:getOperandTypes (Results 1 – 25 of 77) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_traits.h59 for (auto operand_type : op->getOperandTypes()) { in verifyTrait()
91 for (const auto& operand_type : op->getOperandTypes()) { in verifySameOperandsAndResultElementTypeResolveRef()
Dtf_device.cc148 for (auto result_type : region_terminator->getOperandTypes()) { in Verify()
366 for (const auto& type : terminator.getOperandTypes()) in ParseReplicateOp()
503 llvm::enumerate(terminator.getOperandTypes())) { in Verify()
/external/llvm-project/mlir/lib/Dialect/LLVMIR/IR/
DNVVMDialect.cpp97 SmallVector<Type, 12> operand_types(op.getOperandTypes().begin(), in verify()
98 op.getOperandTypes().end()); in verify()
/external/llvm-project/mlir/lib/Interfaces/
DControlFlowInterfaces.cpp197 if (regionReturn->getOperandTypes() != terminator->getOperandTypes()) in verifyTypesAlongControlFlowEdges()
/external/llvm-project/mlir/lib/IR/
DOperation.cpp776 for (auto opType : op->getOperandTypes()) { in verifyOperandsAreSignlessIntegerLike()
785 for (auto opType : op->getOperandTypes()) { in verifyOperandsAreFloatLike()
800 for (auto opType : llvm::drop_begin(op->getOperandTypes(), 1)) in verifySameTypeOperands()
864 for (auto opType : llvm::drop_begin(op->getOperandTypes(), 1)) { in verifySameOperandsShape()
882 for (auto opType : llvm::drop_begin(op->getOperandTypes(), 1)) { in verifySameOperandsAndResultShape()
941 for (auto opType : op->getOperandTypes()) { in verifySameOperandsAndResultType()
1102 llvm::make_filter_range(op->getOperandTypes(), isMappableType)); in verifyElementwiseMappable()
1168 if (llvm::any_of(op->getOperandTypes(), in printOneResultOp()
/external/llvm-project/mlir/lib/Dialect/
DTraits.cpp192 hasTensorOrVectorType(op->getOperandTypes()); in verifyCompatibleOperandBroadcast()
201 op->getOperandTypes(), [](Type t) { return t.isa<RankedTensorType>(); }); in verifyCompatibleOperandBroadcast()
/external/llvm-project/mlir/examples/toy/Ch3/mlir/
DDialect.cpp79 if (llvm::all_of(op->getOperandTypes(), in printBinaryOp()
86 printer.printFunctionalType(op->getOperandTypes(), op->getResultTypes()); in printBinaryOp()
/external/llvm-project/mlir/examples/toy/Ch2/mlir/
DDialect.cpp79 if (llvm::all_of(op->getOperandTypes(), in printBinaryOp()
86 printer.printFunctionalType(op->getOperandTypes(), op->getResultTypes()); in printBinaryOp()
/external/llvm-project/mlir/examples/toy/Ch5/mlir/
DShapeInferencePass.cpp95 return llvm::all_of(op->getOperandTypes(), [](Type operandType) { in allOperandsInferred()
DDialect.cpp135 if (llvm::all_of(op->getOperandTypes(), in printBinaryOp()
142 printer.printFunctionalType(op->getOperandTypes(), op->getResultTypes()); in printBinaryOp()
/external/llvm-project/mlir/examples/toy/Ch4/mlir/
DShapeInferencePass.cpp95 return llvm::all_of(op->getOperandTypes(), [](Type operandType) { in allOperandsInferred()
DDialect.cpp135 if (llvm::all_of(op->getOperandTypes(), in printBinaryOp()
142 printer.printFunctionalType(op->getOperandTypes(), op->getResultTypes()); in printBinaryOp()
/external/llvm-project/mlir/examples/toy/Ch7/mlir/
DShapeInferencePass.cpp95 return llvm::all_of(op->getOperandTypes(), [](Type operandType) { in allOperandsInferred()
/external/llvm-project/mlir/examples/toy/Ch6/mlir/
DShapeInferencePass.cpp95 return llvm::all_of(op->getOperandTypes(), [](Type operandType) { in allOperandsInferred()
DDialect.cpp135 if (llvm::all_of(op->getOperandTypes(), in printBinaryOp()
142 printer.printFunctionalType(op->getOperandTypes(), op->getResultTypes()); in printBinaryOp()
/external/llvm-project/mlir/lib/Dialect/Shape/Transforms/
DStructuralTypeConversions.cpp68 return typeConverter.isLegal(op.getOperandTypes()); in populateShapeStructuralTypeConversionsAndLegality()
/external/llvm-project/mlir/lib/Dialect/Quant/IR/
DQuantOps.cpp74 for (auto input : llvm::zip(op.getOperandTypes(), op.input_specs())) { in verifyRegionOp()
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DElementwiseToLinalg.cpp24 return llvm::all_of(op->getOperandTypes(), in isElementwiseMappableOpOnRankedTensors()
/external/llvm-project/mlir/test/lib/Transforms/
DTestDecomposeCallGraphTypes.cpp41 return typeConverter.isLegal(op.getOperandTypes()); in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize_functional_ops.cc47 auto return_types = llvm::to_vector<4>(terminator->getOperandTypes()); in UpdateFuncType()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dconstant_fold.cc64 int64_t operands_size = get_size(inst->getOperandTypes()); in ShouldBeFolded()
Dshape_inference.cc746 for (auto result : zip(op.getResults(), then_yield->getOperandTypes(), in InferShapeForIfRegion()
747 else_yield->getOperandTypes())) { in InferShapeForIfRegion()
1080 while_region.body().front().getTerminator()->getOperandTypes()); in InferShapeForSingleOperation()
1154 std::equal(caller->getOperandTypes().begin(), in PropagateShapeToFunctions()
1155 caller->getOperandTypes().end(), in PropagateShapeToFunctions()
1156 callers.front()->getOperandTypes().begin()); in PropagateShapeToFunctions()
1519 return_op.getOperandTypes())); in InferShapeForFunctionReturnType()
Dparallel_execute_to_islands.cc110 island_op.getLoc(), yield.getOperandTypes(), in ExpandParallelExecuteToIslands()
/external/llvm-project/mlir/lib/Dialect/SCF/Transforms/
DStructuralTypeConversions.cpp149 return typeConverter.isLegal(op.getOperandTypes()); in populateSCFStructuralTypeConversionsAndLegality()
/external/llvm-project/mlir/lib/Conversion/GPUToVulkan/
DConvertGPULaunchFuncToVulkanLaunchFunc.cpp106 SmallVector<Type, 8> gpuLaunchTypes(launchOp.getOperandTypes()); in declareVulkanLaunchFunc()

1234