#NEWLINE# def forward_gpu( self , x ) : #NEWLINE# #INDENT# ( a , b ) = x #NEWLINE# shape = self . _output_shape ( a , b ) #NEWLINE# ret = cuda . cupy . zeros ( shape , dtype = a . dtype ) #NEWLINE# _batch_matmul_gpu ( a , self , transa = self . transa , transb = self . transb , out = ret ) #NEWLINE# return ( ret , )