@ -49,7 +49,7 @@ class SequenceConvKernel : public framework::OpKernel<T> {
int up_pad = std : : max ( 0 , - context_start ) ;
int down_pad = std : : max ( 0 , context_start + context_length - 1 ) ;
int sequence_width = static_cast < in t> ( in - > dims ( ) [ 1 ] ) ;
auto sequence_width = static_cast < in t64_ t> ( in - > dims ( ) [ 1 ] ) ;
framework : : DDim col_shape = { in - > dims ( ) [ 0 ] ,
context_length * sequence_width } ;
@ -62,7 +62,7 @@ class SequenceConvKernel : public framework::OpKernel<T> {
set_zero ( dev_ctx , & col , static_cast < T > ( 0 ) ) ;
math : : ContextProjectFunctor < DeviceContext , T > seq_project_functor ;
seq_project_functor ( dev_ctx , * in , * padding_data , padding_trainable ,
seq_project_functor ( dev_ctx , * in , padding_data , padding_trainable ,
context_start , context_length , context_stride , up_pad ,
down_pad , & col ) ;
@ -93,7 +93,7 @@ class SequenceConvGradKernel : public framework::OpKernel<T> {
int up_pad = std : : max ( 0 , - context_start ) ;
int down_pad = std : : max ( 0 , context_start + context_length - 1 ) ;
int sequence_width = static_cast < in t> ( in - > dims ( ) [ 1 ] ) ;
auto sequence_width = static_cast < in t64_ t> ( in - > dims ( ) [ 1 ] ) ;
math : : SetConstant < DeviceContext , T > set_zero ;
auto & dev_ctx = context . template device_context < DeviceContext > ( ) ;
@ -144,7 +144,7 @@ class SequenceConvGradKernel : public framework::OpKernel<T> {
padding_data = context . Input < Tensor > ( " PaddingData " ) ;
}
seq_project_functor ( dev_ctx , * in , * padding_data , padding_trainable ,
seq_project_functor ( dev_ctx , * in , padding_data , padding_trainable ,
context_start , context_length , context_stride , up_pad ,
down_pad , & col ) ;