diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2016-10-10 18:34:09 -0800 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-10-10 19:52:24 -0700 |
| commit | 7d0e34ef174ba62897535b9139c0510f3599acad (patch) | |
| tree | b32ee9f95681de513e0bf28a6fe5a033324c4902 /tensorflow/core/kernels/strided_slice_op.cc | |
| parent | 6afe08f24e8889bd808d746ae5d3872543d015f9 (diff) | |
Register kernels for int32.
Change: 135751986
Diffstat (limited to 'tensorflow/core/kernels/strided_slice_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/strided_slice_op.cc | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/strided_slice_op.cc b/tensorflow/core/kernels/strided_slice_op.cc index e0b4396bf0..b86c727829 100644 --- a/tensorflow/core/kernels/strided_slice_op.cc +++ b/tensorflow/core/kernels/strided_slice_op.cc @@ -394,6 +394,17 @@ REGISTER_KERNEL_BUILDER(Name("StridedSlice") .HostMemory("strides") .HostMemory("output"), StridedSliceOp<CPUDevice, int32>); +REGISTER_KERNEL_BUILDER(Name("StridedSliceGrad") + .Device(DEVICE_GPU) + .TypeConstraint<int32>("T") + .TypeConstraint<int32>("Index") + .HostMemory("shape") + .HostMemory("begin") + .HostMemory("end") + .HostMemory("strides") + .HostMemory("dy") + .HostMemory("output"), + StridedSliceGradOp<CPUDevice, int32>); #undef REGISTER_GPU |