Lines Matching defs:memorySpace

196   std::optional<Attribute> memorySpace = copyBufferType->getMemorySpace();
197   if (!memorySpace)
198     memorySpace = options.defaultMemorySpaceFn(tensorType);
199   if (memorySpace.has_value())
200     allocTensorOp.setMemorySpaceAttr(memorySpace.value());
316 defaultFunctionArgTypeConverter(TensorType type, Attribute memorySpace,
319   return getMemRefTypeWithFullyDynamicLayout(type, memorySpace);
323 defaultUnknownTypeConverter(Value value, Attribute memorySpace,
326       llvm::cast<TensorType>(value.getType()), memorySpace);
363   functionArgTypeConverterFn = [=](TensorType tensorType, Attribute memorySpace,
368                                                                   memorySpace);
370                                                               memorySpace);
775                                             Attribute memorySpace) {
783                                    memorySpace);
791                            memorySpace);
794   return options.unknownTypeConverterFn(value, memorySpace, options);
799                                                    Attribute memorySpace) {
804                                    memorySpace);
816                          memorySpace);
823                                                      Attribute memorySpace) {
828                                    memorySpace);
836                          memorySpace);