rnn_backward(mkldnn.rnn_backward.primitive_desc aprimitive_desc,
mkldnn.primitive.at src_layer,
mkldnn.primitive.at src_iter,
mkldnn.primitive.at weights_layer,
mkldnn.primitive.at weights_iter,
mkldnn.primitive.at bias,
mkldnn.primitive.at dst_layer,
mkldnn.primitive.at dst_iter,
mkldnn.memory diff_src_layer,
mkldnn.memory diff_src_iter,
mkldnn.memory diff_weights_layer,
mkldnn.memory diff_weights_iter,
mkldnn.memory diff_bias,
mkldnn.primitive.at diff_dst_layer,
mkldnn.primitive.at diff_dst_iter,
mkldnn.primitive.at workspace) |