Lines Matching defs:shape

23                       SmallVector<int64_t> &shape) {
24   SmallVector<int64_t> old = shape;
26     shape[i] = old[trans[i]];
45   SmallVector<int64_t> shape;
47     shape = SmallVector<int64_t>(ty.getShape());
49     shape.push_back(1);
50   return shape;
79 // - tensor descriptor has a sg_map attribute and the distributed vector shape
80 //   matches the tensor descriptor shape when scaled using sg_map factors on
123         ValueRange({}) /* empty dynamic shape */,
125         staticOffsets /* const offsets */, {} /* empty const shape*/,
132                            llvm::ArrayRef<OpFoldResult> shape,
134   assert(shape.size() && offsets.size() && strides.size() &&
135          shape.size() == strides.size() && shape.size() == offsets.size());
145   dispatchIndexOpFoldResults(shape, dynamicShape, staticShape);
159                            llvm::ArrayRef<OpFoldResult> shape,
161   assert(shape.size() && offsets.size() && strides.size() &&
162          shape.size() == strides.size() && shape.size() == offsets.size());
172   dispatchIndexOpFoldResults(shape, dynamicShape, staticShape);
207   // mismatches among shape, strides, and offsets are
212         "Expecting the rank of shape, strides, offsets, and source (if source "
221         "ranks of shape, strides, offsets or the memref source.");
320     return emitOpError() << "Result shape doesn't match TensorDesc shape."
321                          << "The expected shape is " << makeString(tdescShape)
322                          << ". But the given shape is "
357     return emitOpError() << "Result shape doesn't match TensorDesc shape."
358                          << "The expected shape is " << makeString(tdescShape)
359                          << ". But the given shape is "
449   SmallVector<int64_t> shape({(int64_t)getNumOffsets()});
451     shape.push_back(chunkSize);
454   if (shape != tdescShape)
455     return emitOpError("Incorrect TensorDesc shape. ")
456            << "Expected is " << makeString(shape) << "\n";
521     return emitOpError("Unexpected result shape")
522            << "(Expected shape: " << makeString(tdescShape)
523            << ", Given shape: " << makeString(valueShape) << ").\n";
560     return emitOpError("Unexpected value shape")
561            << "(Expected shape: " << makeString(tdescShape)
562            << ", Given shape: " << makeString(valueShape) << ").\n";