File tree Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Expand file tree Collapse file tree 1 file changed +1
-5
lines changed Original file line number Diff line number Diff line change @@ -735,10 +735,7 @@ void FillPhiloxRandom<SYCLDevice, Distribution>::operator()(
735735 PhiloxRandomOp< \
736736 SYCLDevice, \
737737 random::TruncatedNormalDistribution< \
738- random::SingleSampleAdapter<random::PhiloxRandom>, TYPE> >); \
739- REGISTER_KERNEL_BUILDER ( \
740- Name (" RandomGamma" ).Device(DEVICE_SYCL).TypeConstraint<TYPE>(" T" ), \
741- RandomGammaOp<TYPE>)
738+ random::SingleSampleAdapter<random::PhiloxRandom>, TYPE> >);
742739
743740#define REGISTER_INT (IntType ) \
744741 REGISTER_KERNEL_BUILDER (Name(" RandomUniformInt" ) \
@@ -749,7 +746,6 @@ void FillPhiloxRandom<SYCLDevice, Distribution>::operator()(
749746 .TypeConstraint<IntType>(" Tout" ), \
750747 RandomUniformIntOp<SYCLDevice, IntType>);
751748
752- TF_CALL_half (REGISTER);
753749TF_CALL_float (REGISTER);
754750TF_CALL_double (REGISTER);
755751TF_CALL_int32 (REGISTER_INT);
You can’t perform that action at this time.
0 commit comments