Home
last modified time | relevance | path

Searched defs:BlockMask (Results 1 – 10 of 10) sorted by relevance

/external/pytorch/docs/source/
Dnn.attention.flex_attention.rst21 BlockMask chapter
/external/swiftshader/third_party/llvm-16.0/llvm/lib/Target/ARM/
DMVEVPTBlockPass.cpp186 ARM::PredBlockMask BlockMask = GetInitialBlockMask(BlockSize); in CreateVPTBlock() local
269 ARM::PredBlockMask BlockMask = in InsertVPTBlocks() local
DThumb2InstrInfo.cpp819 ARM::PredBlockMask BlockMask = ARM::PredBlockMask::T; in recomputeVPTBlockMask() local
/external/swiftshader/third_party/llvm-16.0/llvm/lib/Target/ARM/Utils/
DARMBaseInfo.cpp18 ARM::PredBlockMask expandPredBlockMask(ARM::PredBlockMask BlockMask, in expandPredBlockMask()
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Target/ARM/
DMVEVPTBlockPass.cpp139 unsigned BlockMask = getARMVPTBlockMask(VPTInstCnt); in InsertVPTBlocks() local
/external/pytorch/torch/nn/attention/
Dflex_attention.py191 class BlockMask: class
921 block_mask: Optional[BlockMask] = None,
/external/pytorch/test/inductor/
Dtest_flex_decoding.py268 block_mask: Optional[BlockMask] = None,
/external/llvm/lib/Transforms/Vectorize/
DLoopVectorize.cpp3946 VectorParts BlockMask = getVectorValue(Zero); in createBlockInMask() local
/external/swiftshader/third_party/llvm-10.0/llvm/lib/Transforms/Vectorize/
DLoopVectorize.cpp6744 VPValue *BlockMask = nullptr; in createBlockInMask() local
/external/swiftshader/third_party/llvm-16.0/llvm/lib/Transforms/Vectorize/
DLoopVectorize.cpp8096 VPValue *BlockMask = nullptr; in createBlockInMask() local