Home
last modified time | relevance | path

Searched refs:ExtractOp (Results 1 – 15 of 15) sorted by relevance

/external/llvm-project/mlir/lib/Conversion/VectorToSPIRV/
DVectorToSPIRV.cpp48 : public SPIRVOpLowering<vector::ExtractOp> {
49 using SPIRVOpLowering<vector::ExtractOp>::SPIRVOpLowering;
51 matchAndRewrite(vector::ExtractOp extractOp, ArrayRef<Value> operands, in matchAndRewrite()
56 vector::ExtractOp::Adaptor adaptor(operands); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlegalize_control_flow.cc87 auto cond_value = builder.create<mlir::tensor::ExtractOp>(loc, if_op.pred()); in LowerIfOp()
171 builder.create<mlir::tensor::ExtractOp>(loc, return_value); in LowerWhileOp()
Dmhlo_control_flow_to_scf.cc123 return b.create<tensor::ExtractOp>( in MatchAndRewrite()
Dhlo_legalize_to_lhlo.cc66 rewriter->create<tensor::ExtractOp>(loc, shape_operand, index); in InsertDynamicAllocAndDealloc()
319 b->create<tensor::ExtractOp>(loc, op.output_dimensions(), i_val); in InsertDynamicMemrefCastOp()
Dlegalize_to_linalg.cc508 dyn_dims.push_back(rewriter.create<tensor::ExtractOp>(loc, shape, index)); in matchAndRewrite()
1326 init_value = rewriter.create<tensor::ExtractOp>(loc, init_value); in matchAndRewrite()
/external/llvm-project/mlir/include/mlir/Dialect/Vector/EDSC/
DIntrinsics.h19 using vector_extract = ValueBuilder<vector::ExtractOp>;
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorTransforms.cpp119 return rewriter.create<vector::ExtractOp>(loc, lowType, val, posAttr); in reshapeLoad()
128 Value ext = rewriter.create<vector::ExtractOp>(loc, vType, val, posAttr); in reshapeLoad()
155 Value ext = rewriter.create<vector::ExtractOp>(loc, vType, result, posAttr); in reshapeStore()
156 Value ins = rewriter.create<vector::ExtractOp>(loc, insType, val, posAttr); in reshapeStore()
1188 Value ext = rewriter.create<vector::ExtractOp>(loc, op.source(), 0); in matchAndRewrite()
1214 Value ext = rewriter.create<vector::ExtractOp>(loc, op.source(), 0); in matchAndRewrite()
1221 Value ext = rewriter.create<vector::ExtractOp>(loc, op.source(), d); in matchAndRewrite()
1298 Value e = rewriter.create<vector::ExtractOp>(loc, eltType, input, ridx); in expandIndices()
1353 Value x = rewriter.create<vector::ExtractOp>(loc, eltType, op.lhs(), pos); in matchAndRewrite()
1357 r = rewriter.create<vector::ExtractOp>(loc, rhsType, acc, pos); in matchAndRewrite()
[all …]
DVectorOps.cpp570 void vector::ExtractOp::build(OpBuilder &builder, OperationState &result, in build()
580 void vector::ExtractOp::build(OpBuilder &builder, OperationState &result, in build()
589 static void print(OpAsmPrinter &p, vector::ExtractOp op) { in print()
624 static LogicalResult verify(vector::ExtractOp op) { in verify()
652 static LogicalResult foldExtractOpFromExtractChain(ExtractOp extractOp) { in foldExtractOpFromExtractChain()
653 if (!extractOp.vector().getDefiningOp<ExtractOp>()) in foldExtractOpFromExtractChain()
657 ExtractOp currentOp = extractOp; in foldExtractOpFromExtractChain()
660 while (ExtractOp nextOp = currentOp.vector().getDefiningOp<ExtractOp>()) { in foldExtractOpFromExtractChain()
669 extractOp.setAttr(ExtractOp::getPositionAttrName(), in foldExtractOpFromExtractChain()
675 static LogicalResult foldExtractOpFromTranspose(ExtractOp extractOp) { in foldExtractOpFromTranspose()
[all …]
/external/llvm-project/mlir/lib/Conversion/VectorToLLVM/
DConvertVectorToLLVM.cpp84 return rewriter.create<ExtractOp>(loc, vector, offset); in extractOne()
738 : ConvertToLLVMPattern(vector::ExtractOp::getOperationName(), context, in VectorExtractOpConversion()
746 auto extractOp = cast<vector::ExtractOp>(op); in matchAndRewrite()
953 Value extrLHS = rewriter.create<ExtractOp>(loc, op.lhs(), i); in matchAndRewrite()
954 Value extrRHS = rewriter.create<ExtractOp>(loc, op.rhs(), i); in matchAndRewrite()
955 Value extrACC = rewriter.create<ExtractOp>(loc, op.acc(), i); in matchAndRewrite()
1002 rewriter.create<ExtractOp>(loc, op.dest(), in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dbufferize_pass.cc174 target.addIllegalOp<tensor::GenerateOp, tensor::ExtractOp, in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dfunctional_control_flow_to_cfg.cc46 auto scalar = builder->create<tensor::ExtractOp>(loc, zero_d); in LowerCondition()
/external/llvm/lib/Target/ARM/
DARMInstrNEON.td2051 PatFrag StoreOp, SDNode ExtractOp, Operand AddrMode>
2055 [(StoreOp (ExtractOp (Ty DPR:$Vd), imm:$lane), AddrMode:$Rn)]> {
2059 class VST1QLNPseudo<ValueType Ty, PatFrag StoreOp, SDNode ExtractOp>
2061 let Pattern = [(StoreOp (ExtractOp (Ty QPR:$src), imm:$lane),
2092 PatFrag StoreOp, SDNode ExtractOp, Operand AdrMode>
2098 [(set GPR:$wb, (StoreOp (ExtractOp (Ty DPR:$Vd), imm:$lane),
2102 class VST1QLNWBPseudo<ValueType Ty, PatFrag StoreOp, SDNode ExtractOp>
2104 let Pattern = [(set GPR:$wb, (StoreOp (ExtractOp (Ty QPR:$src), imm:$lane),
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/ARM/
DARMInstrNEON.td2110 PatFrag StoreOp, SDNode ExtractOp, Operand AddrMode>
2114 [(StoreOp (ExtractOp (Ty DPR:$Vd), imm:$lane), AddrMode:$Rn)]>,
2119 class VST1QLNPseudo<ValueType Ty, PatFrag StoreOp, SDNode ExtractOp>
2121 let Pattern = [(StoreOp (ExtractOp (Ty QPR:$src), imm:$lane),
2159 PatFrag StoreOp, SDNode ExtractOp, Operand AdrMode>
2165 [(set GPR:$wb, (StoreOp (ExtractOp (Ty DPR:$Vd), imm:$lane),
2170 class VST1QLNWBPseudo<ValueType Ty, PatFrag StoreOp, SDNode ExtractOp>
2172 let Pattern = [(set GPR:$wb, (StoreOp (ExtractOp (Ty QPR:$src), imm:$lane),
/external/llvm-project/llvm/lib/Target/ARM/
DARMInstrNEON.td2092 PatFrag StoreOp, SDNode ExtractOp, Operand AddrMode>
2096 [(StoreOp (ExtractOp (Ty DPR:$Vd), imm:$lane), AddrMode:$Rn)]>,
2101 class VST1QLNPseudo<ValueType Ty, PatFrag StoreOp, SDNode ExtractOp>
2103 let Pattern = [(StoreOp (ExtractOp (Ty QPR:$src), imm:$lane),
2141 PatFrag StoreOp, SDNode ExtractOp, Operand AdrMode>
2147 [(set GPR:$wb, (StoreOp (ExtractOp (Ty DPR:$Vd), imm:$lane),
2152 class VST1QLNWBPseudo<ValueType Ty, PatFrag StoreOp, SDNode ExtractOp>
2154 let Pattern = [(set GPR:$wb, (StoreOp (ExtractOp (Ty QPR:$src), imm:$lane),
/external/vixl/src/aarch64/
Dconstants-aarch64.h830 enum ExtractOp { enum