Lines Matching refs:memorySpace
261 defaultUnknownTypeConverter(Value value, unsigned memorySpace, in defaultUnknownTypeConverter() argument
264 memorySpace); in defaultUnknownTypeConverter()
589 Optional<unsigned> memorySpace = None; in getBufferType() local
597 memorySpace = *queriedMemorySpace; in getBufferType()
604 if (!memorySpace.has_value()) in getBufferType()
605 memorySpace = options.defaultMemorySpace; in getBufferType()
608 if (!memorySpace.has_value()) in getBufferType()
611 return getMemRefType(value, options, /*layout=*/{}, *memorySpace); in getBufferType()
700 unsigned memorySpace) { in getMemRefType() argument
703 IntegerType::get(tensorType.getContext(), 64), memorySpace); in getMemRefType()
720 return options.unknownTypeConverterFn(value, memorySpace, options); in getMemRefType()
725 unsigned memorySpace) { in getMemRefTypeWithFullyDynamicLayout() argument
729 memorySpace); in getMemRefTypeWithFullyDynamicLayout()
734 IntegerType::get(tensorType.getContext(), 64), memorySpace); in getMemRefTypeWithFullyDynamicLayout()
750 unsigned memorySpace) { in getMemRefTypeWithStaticIdentityLayout() argument
754 memorySpace); in getMemRefTypeWithStaticIdentityLayout()
760 IntegerType::get(tensorType.getContext(), 64), memorySpace); in getMemRefTypeWithStaticIdentityLayout()