Lines Matching defs:vectorShape

611   /// Perform a distributed load of a vector operand of `vectorShape` for a
615   /// The `vectorShape` matches existing NVGPU dialect op specification but
620                                       ArrayRef<int64_t> vectorShape);
630   /// Perform a distributed store of a vector operand of `vectorShape` for a
634   /// The `vectorShape` matches existing NVGPU dialect op specification but
638       Value memref, IndexCalculator indexFn, ArrayRef<int64_t> vectorShape);
655   auto vectorShape = vectorType.getShape();
656   auto strides = computeStrides(vectorShape);
657   for (int64_t idx = 0, e = vectorShape[0] * strides[0]; idx < e; ++idx) {
683     IndexCalculator indexFn, ArrayRef<int64_t> vectorShape) {
687   auto vt = VectorType::get(vectorShape, elementType);
723     Value memref, IndexCalculator indexFn, ArrayRef<int64_t> vectorShape) {