removing repeated hostcast lines

Change: 115472914
This commit is contained in:
A. Unique TensorFlower 2016-02-24 11:18:29 -08:00 committed by TensorFlower Gardener
parent 57df84c47e
commit 87a289103f

View File

@ -77,9 +77,5 @@ class BitcastOp : public OpKernel {
};
REGISTER_KERNEL_BUILDER(Name("Bitcast").Device(DEVICE_CPU), BitcastOp);
REGISTER_KERNEL_BUILDER(Name("_HostCast").Device(DEVICE_CPU), BitcastOp);
REGISTER_KERNEL_BUILDER(
Name("_HostCast").Device(DEVICE_GPU).HostMemory("x").HostMemory("y"),
BitcastOp);
} // end namespace tensorflow