parameters: (self)
return_statement: return bprop
docstring: brop of the matmulDDS operator
docstring_summary: brop of the matmulDDS operator
docstring_tokens: brop of the matmulDDS operator
function: def get_bprop(self):
      """brop of the matmulDDS operator"""
      def bprop(q, k, local_mask, global_mask, out, d_out):
          lc, gc = out
          d_lc, d_gc = d_out
          dq, dk = inner.MatmulDDSGrad()(q, k, lc, gc, d_lc, d_gc)
          dk = P.Transpose()(dk, (1, 0, 3, 2))
          all_d = (dq, dk, zeros_like(local_mask), zeros_like(global_mask))
          return all_d
      return bprop
function_tokens: def get_bprop ( self ) : def bprop ( q , k , local_mask , global_mask , out , d_out ) : lc , gc = out d_lc , d_gc = d_out dq , dk = inner . MatmulDDSGrad ( ) ( q , k , lc , gc , d_lc , d_gc ) dk = P . Transpose ( ) ( dk , ( 1 , 0 , 3 , 2 ) ) all_d = ( dq , dk , zeros_like ( local_mask ) , zeros_like ( global_mask ) ) return all_d return bprop