def local_inplace_sparse_block_gemv(node): """ SparseBlockGemv(inplace=False) -> SparseBlockGemv(inplace=True) """ if isinstance(node.op, SparseBlockGemv) and not node.op.inplace: new_node = sparse_block_gemv_inplace(*node.inputs) return [new_node] return False
def local_inplace_sparse_block_gemv(fgraph, node): """ SparseBlockGemv(inplace=False) -> SparseBlockGemv(inplace=True) """ 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