File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -345,7 +345,7 @@ class Pad2dCPUKernel : public framework::OpKernel<T> {
345
345
GetPaddings (pads, context);
346
346
auto mode = context.Attr <std::string>(" mode" );
347
347
auto data_format = context.Attr <std::string>(" data_format" );
348
- T value = context.Attr <T >(" pad_value" );
348
+ T value = static_cast <T>( context.Attr <float >(" pad_value" ) );
349
349
350
350
auto * x = context.Input <Tensor>(" X" );
351
351
auto in_dims = x->dims ();
Original file line number Diff line number Diff line change @@ -314,7 +314,7 @@ class Pad2dCUDAKernel : public framework::OpKernel<T> {
314
314
GetPaddings (pads, context);
315
315
auto mode = context.Attr <std::string>(" mode" );
316
316
auto data_format = context.Attr <std::string>(" data_format" );
317
- T value = context.Attr <T >(" pad_value" );
317
+ T value = static_cast <T>( context.Attr <float >(" pad_value" ) );
318
318
319
319
auto * x = context.Input <Tensor>(" X" );
320
320
auto in_dims = x->dims ();
You can’t perform that action at this time.
0 commit comments