| /external/llvm-project/mlir/include/mlir/Transforms/ |
| D | DialectConversion.h | 574 template <typename OpT> void setOpAction(LegalizationAction action) { in setOpAction()
575 setOpAction(OperationName(OpT::getOperationName(), &ctx), action); in setOpAction()
579 template <typename OpT> void addLegalOp() { in addLegalOp()
580 setOpAction<OpT>(LegalizationAction::Legal); in addLegalOp()
582 template <typename OpT, typename OpT2, typename... OpTs> void addLegalOp() { in addLegalOp()
583 addLegalOp<OpT>(); in addLegalOp()
589 template <typename OpT> void addDynamicallyLegalOp() { in addDynamicallyLegalOp()
590 setOpAction<OpT>(LegalizationAction::Dynamic); in addDynamicallyLegalOp()
592 template <typename OpT, typename OpT2, typename... OpTs>
594 addDynamicallyLegalOp<OpT>(); in addDynamicallyLegalOp()
[all …]
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Block.h | 172 template <typename OpT>
173 using op_iterator = detail::op_iterator<OpT, iterator>;
177 template <typename OpT> iterator_range<op_iterator<OpT>> getOps() { in getOps()
179 return {detail::op_filter_iterator<OpT, iterator>(begin(), endIt), in getOps()
180 detail::op_filter_iterator<OpT, iterator>(endIt, endIt)}; in getOps()
182 template <typename OpT> op_iterator<OpT> op_begin() { in op_begin()
183 return detail::op_filter_iterator<OpT, iterator>(begin(), end()); in op_begin()
185 template <typename OpT> op_iterator<OpT> op_end() { in op_end()
186 return detail::op_filter_iterator<OpT, iterator>(end(), end()); in op_end()
|
| D | BlockSupport.h | 126 template <typename OpT, typename IteratorT>
129 static bool filter(Operation &op) { return llvm::isa<OpT>(op); } in filter()
142 template <typename OpT, typename IteratorT>
144 : public llvm::mapped_iterator<op_filter_iterator<OpT, IteratorT>,
145 OpT (*)(Operation &)> {
146 static OpT unwrap(Operation &op) { return cast<OpT>(op); } in unwrap()
149 using reference = OpT;
152 op_iterator(op_filter_iterator<OpT, IteratorT> it) in op_iterator() argument
153 : llvm::mapped_iterator<op_filter_iterator<OpT, IteratorT>, in op_iterator()
154 OpT (*)(Operation &)>(it, &unwrap) {} in op_iterator()
|
| D | Region.h | 152 template <typename OpT>
153 using op_iterator = detail::op_iterator<OpT, OpIterator>;
163 template <typename OpT> op_iterator<OpT> op_begin() { in op_begin()
164 return detail::op_filter_iterator<OpT, OpIterator>(op_begin(), op_end()); in op_begin()
166 template <typename OpT> op_iterator<OpT> op_end() { in op_end()
167 return detail::op_filter_iterator<OpT, OpIterator>(op_end(), op_end()); in op_end()
169 template <typename OpT> iterator_range<op_iterator<OpT>> getOps() { in getOps()
171 return {detail::op_filter_iterator<OpT, OpIterator>(op_begin(), endIt), in getOps()
172 detail::op_filter_iterator<OpT, OpIterator>(endIt, endIt)}; in getOps()
|
| D | OpDefinition.h | 507 template <typename OpT>
509 return getRegion().template getOps<OpT>(); in getOps()
880 template <typename OpT, typename T = void>
882 typename std::enable_if_t<OpT::template hasTrait<OneRegion>(), T>;
884 template <typename OpT = ConcreteType>
885 enable_if_single_region<OpT, Block::iterator> begin() { in begin()
888 template <typename OpT = ConcreteType>
889 enable_if_single_region<OpT, Block::iterator> end() { in end()
892 template <typename OpT = ConcreteType>
893 enable_if_single_region<OpT, Operation &> front() { in front()
[all …]
|
| D | Builders.h | 474 template <typename OpT> OpT cloneWithoutRegions(OpT op) { in cloneWithoutRegions()
475 return cast<OpT>(cloneWithoutRegions(*op.getOperation())); in cloneWithoutRegions()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | rewrite_tpu_embedding_ops.cc | 36 template <typename OpT>
37 OpT AddOperandAndRewriteAs(Operation* op, Value operand, OpBuilder* builder) { in AddOperandAndRewriteAs()
41 auto new_op = builder->create<OpT>(op->getLoc(), op->getResultTypes(), in AddOperandAndRewriteAs()
51 template <typename OpT>
52 LogicalResult GetOp(FuncOp func, OpT* result) { in GetOp()
54 for (auto op : func.getOps<OpT>()) { in GetOp()
|
| D | lower_tf.cc | 241 template <typename OpT>
246 OpT::getOperationName(), in LowerDynamicStitchOp()
253 auto op = cast<OpT>(src_op); in matchAndRewrite()
|
| /external/llvm-project/mlir/include/mlir/Pass/ |
| D | AnalysisManager.h | 138 template <typename AnalysisT, typename OpT>
139 typename std::enable_if<std::is_constructible<AnalysisT, OpT>::value,
142 return getAnalysisImpl<AnalysisT, OpT>(pi, cast<OpT>(ir)); in getAnalysis()
172 template <typename AnalysisT, typename OpT>
173 AnalysisT &getAnalysisImpl(PassInstrumentor *pi, OpT op) { in getAnalysisImpl()
255 template <typename AnalysisT, typename OpT>
257 return impl->analyses.getAnalysis<AnalysisT, OpT>(getPassInstrumentor()); in getAnalysis()
273 template <typename AnalysisT, typename OpT>
274 AnalysisT &getChildAnalysis(OpT child) { in getChildAnalysis()
275 return nest(child).template getAnalysis<AnalysisT, OpT>(); in getChildAnalysis()
|
| D | Pass.h | 196 template <typename AnalysisT, typename OpT>
198 return getAnalysisManager().getAnalysis<AnalysisT, OpT>(); in getAnalysis()
315 template <typename OpT = void> class OperationPass : public Pass {
317 OperationPass(TypeID passID) : Pass(passID, OpT::getOperationName()) {} in OperationPass()
321 return pass->getOpName() == OpT::getOperationName(); in classof()
325 OpT getOperation() { return cast<OpT>(Pass::getOperation()); } in getOperation()
331 return Pass::getAnalysis<AnalysisT, OpT>(); in getAnalysis()
|
| D | PassManager.h | 77 template <typename OpT> OpPassManager &nest() { in nest()
78 return nest(OpT::getOperationName()); in nest()
87 template <typename OpT> void addNestedPass(std::unique_ptr<Pass> pass) { in addNestedPass()
88 nest<OpT>().addPass(std::move(pass)); in addNestedPass()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_collective_thunk.h | 77 template <typename OpT>
78 NcclCollectiveConfig GetNcclCollectiveConfigForMlir(OpT op, in GetNcclCollectiveConfigForMlir()
|
| D | ir_emitter_unnested.h | 216 template <typename ThunkType, typename OpT>
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | lhlo_ops.cc | 65 template <typename OpT>
66 LogicalResult VerifyReplicaGroups(OpT op, bool is_uniform_sized) { in VerifyReplicaGroups()
|
| D | hlo_ops.cc | 92 template <typename OpT>
93 static LogicalResult VerifyDimAttr(OpT op) { in VerifyDimAttr()
|
| /external/llvm-project/mlir/include/mlir/Conversion/PDLToPDLInterp/ |
| D | PDLToPDLInterp.h | 20 template <typename OpT>
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_ops_a_m.cc | 165 template <typename OpT,
167 OpT, BatchMatMulOp, BatchMatMulV2Op>::value>::type * = nullptr>
168 static LogicalResult Verify(OpT op) { in Verify()
194 if (std::is_same<OpT, BatchMatMulOp>()) { in Verify()
942 template <typename OpT,
944 OpT, ConcatOp, ConcatV2Op>::value>::type * = nullptr>
945 static LogicalResult Verify(OpT op) { in Verify()
949 int axis_idx = std::is_same<OpT, ConcatOp>() ? 0 : 1; in Verify()
1284 template <typename OpT, typename std::enable_if<llvm::is_one_of<
1285 OpT, CumsumOp, CumprodOp>::value>::type * = nullptr>
[all …]
|
| D | tf_ops_helpers.inc | 451 typename OpT,
453 OpT, AddV2Op, SubOp, MulOp, DivOp, RealDivOp>::value>::type * = nullptr>
454 OpFoldResult IdentityArithmeticOpFolder(OpT arithmetic_op,
481 (std::is_same<OpT, AddV2Op>::value || std::is_same<OpT, MulOp>::value);
489 (std::is_same<OpT, MulOp>::value || std::is_same<OpT, DivOp>::value ||
490 std::is_same<OpT, RealDivOp>::value)
|
| /external/llvm-project/mlir/include/mlir/Conversion/VectorToROCDL/ |
| D | VectorToROCDL.h | 17 template <typename OpT>
|
| /external/llvm-project/mlir/include/mlir/Conversion/GPUToROCDL/ |
| D | GPUToROCDLPass.h | 18 template <typename OpT>
|
| /external/llvm-project/mlir/include/mlir/Conversion/GPUToNVVM/ |
| D | GPUToNVVMPass.h | 19 template <typename OpT> class OperationPass;
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | lower_static_tensor_list.cc | 315 template <typename OpT>
316 struct ConvertTensorListInitOp : public OpConversionPattern<OpT> {
317 using OpConversionPattern<OpT>::OpConversionPattern;
321 virtual Value GetNumElements(OpT op, ArrayRef<Value> operands,
328 OpT op, ArrayRef<Value> operands, in matchAndRewrite()
|
| /external/llvm-project/mlir/docs/ |
| D | Interfaces.md | 139 template <typename OpT>
143 return llvm::cast<OpT>(op).getNumInputs();
|
| /external/llvm-project/mlir/lib/Conversion/PDLToPDLInterp/ |
| D | PDLToPDLInterp.cpp | 334 template <typename OpT, typename PredT, typename ValT = typename PredT::KeyTy>
345 builder.create<OpT>(val.getLoc(), val, values, defaultDest, blocks); in createSwitchOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1024 template <typename OpT>
1025 void SetupChannelIdAttribute(OpT op, const xla::HloChannelInstruction* instr, in SetupChannelIdAttribute()
1034 template <typename OpT>
1035 Status SetupCommonCollectiveOpAttributes(OpT op, const HloInstruction* instr, in SetupCommonCollectiveOpAttributes()
|