@gof.local_optimizer([SparseBlockGemv], inplace=True)
def local_inplace_sparse_block_gemv(node):
 if (isinstance(node.op, SparseBlockGemv) and (not node.op.inplace)):
  new_node = sparse_block_gemv_inplace(*node.inputs)
  copy_stack_trace(node.outputs[0], new_node)
  return [new_node]
 return False

@ gof . local_optimizer ( [ SparseBlockGemv ] , inplace = True ) def local_inplace_sparse_block_gemv ( node ) : if ( isinstance ( node . op , SparseBlockGemv ) and ( not node . op . inplace ) ) : new_node = sparse_block_gemv_inplace ( * node . inputs ) copy_stack_trace ( node . outputs [ 0 ] , new_node ) return [ new_node ] return False
sparseblockgemv -> sparseblockgemv .