Searched refs:SizesArray (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 386 llvm::ArrayRef<int64_t> SizesArray() const { return sizes; } in SizesArray() function
472 lhs_dot_dimensions_info.batch_dimensions().SizesArray(), in ConvertDot()
473 lhs_dot_dimensions_info.out_dimensions().SizesArray(), in ConvertDot()
474 lhs_dot_dimensions_info.contracting_dimensions().SizesArray()); in ConvertDot()
490 rhs_dot_dimensions_info.batch_dimensions().SizesArray(), in ConvertDot()
491 rhs_dot_dimensions_info.contracting_dimensions().SizesArray(), in ConvertDot()
492 rhs_dot_dimensions_info.out_dimensions().SizesArray()); in ConvertDot()
503 lhs_dot_dimensions_info.batch_dimensions().SizesArray(), in ConvertDot()
515 rhs_dot_dimensions_info.batch_dimensions().SizesArray(), in ConvertDot()
527 Concat<int64_t>(lhs_dot_dimensions_info.batch_dimensions().SizesArray(), in ConvertDot()
|
| /external/llvm-project/polly/lib/Exchange/ |
| D | JSONExporter.cpp | 576 const json::Array &SizesArray = *Array.getArray("sizes"); in areArraysEqual() local
577 if (RawStringOstream.str() != SizesArray[i].getAsString().getValue()) in areArraysEqual()
661 const json::Array &SizesArray = *Array.getArray("sizes"); in importArrays() local
663 for (unsigned i = 0; i < SizesArray.size(); i++) { in importArrays()
664 auto Size = std::stoi(SizesArray[i].getAsString().getValue().str()); in importArrays()
|
| /external/clang/lib/CodeGen/ |
| D | CGOpenMPRuntime.cpp | 5552 llvm::Value *&PointersArray, llvm::Value *&SizesArray, in emitOffloadingArrays() argument
5561 BasePointersArray = PointersArray = SizesArray = MapTypesArray = nullptr; in emitOffloadingArrays()
5590 SizesArray = in emitOffloadingArrays()
5606 SizesArray = SizesArrayGbl; in emitOffloadingArrays()
5651 llvm::ArrayType::get(CGM.SizeTy, PointerNumVal), SizesArray, in emitOffloadingArrays()
5668 llvm::Value *PointersArray, llvm::Value *SizesArray, in emitOffloadingArraysArgument() argument
5680 llvm::ArrayType::get(CGM.SizeTy, NumElems), SizesArray, in emitOffloadingArraysArgument()
5789 llvm::Value *SizesArray; in emitTargetCall() local
5791 emitOffloadingArrays(CGF, BasePointersArray, PointersArray, SizesArray, in emitTargetCall()
5795 SizesArray, MapTypesArray, BasePointersArray, in emitTargetCall()
[all …]
|
| /external/llvm-project/clang/lib/CodeGen/ |
| D | CGOpenMPRuntime.h | 1640 llvm::Value *SizesArray = nullptr; variable
1671 SizesArray = nullptr; in clearArrayInfo()
1681 return BasePointersArray && PointersArray && SizesArray && in isValid()
|
| D | CodeGenFunction.h | 3375 Address SizesArray = Address::invalid();
3380 Address SizesArray, Address MappersArray,
3383 SizesArray(SizesArray), MappersArray(MappersArray),
|
| D | CGOpenMPRuntime.cpp | 9186 Info.SizesArray = in emitOffloadingArrays()
9210 Info.SizesArray = SizesArrayGbl; in emitOffloadingArrays()
9304 Info.SizesArray, in emitOffloadingArrays()
9367 llvm::ArrayType::get(CGM.Int64Ty, Info.NumberOfPtrs), Info.SizesArray, in emitOffloadingArraysArgument()
10016 InputInfo.SizesArray.getPointer(), in emitTargetCall()
10035 InputInfo.SizesArray.getPointer(), in emitTargetCall()
10154 CGF, Info.BasePointersArray, Info.PointersArray, Info.SizesArray, in emitTargetCall()
10163 InputInfo.SizesArray = Address(Info.SizesArray, CGM.getPointerAlign()); in emitTargetCall()
10929 InputInfo.SizesArray.getPointer(), in emitTargetDataStandAloneCall()
11040 CGF, Info.BasePointersArray, Info.PointersArray, Info.SizesArray, in emitTargetDataStandAloneCall()
[all …]
|
| D | CGStmtOpenMP.cpp | 4218 [&InputInfo]() { return InputInfo.SizesArray; }); in EmitOMPTargetTaskBasedDirective()
4275 InputInfo.SizesArray = CGF.Builder.CreateConstArrayGEP( in EmitOMPTargetTaskBasedDirective()
|
| /external/llvm-project/llvm/lib/Transforms/IPO/ |
| D | OpenMPOpt.cpp | 1032 auto *SizesArray = cast<AllocaInst>(V); in getValuesInOffloadArrays() local
1033 if (!OAs[2].initialize(*SizesArray, RuntimeCall)) in getValuesInOffloadArrays()
|