Skip to content

Commit f113ef0

Browse files
Luke Iwanskibenoitsteiner
authored andcommitted
[OpenCL] RandomGamma has no GPU friendly implementation (#57)
1 parent 8ed9148 commit f113ef0

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

tensorflow/core/kernels/random_op.cc

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff 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);
753749
TF_CALL_float(REGISTER);
754750
TF_CALL_double(REGISTER);
755751
TF_CALL_int32(REGISTER_INT);

0 commit comments

Comments
 (0)