removing repeated hostcast lines
Change: 115472914
This commit is contained in:
parent
57df84c47e
commit
87a289103f
@ -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
|
||||
|
Loading…
x
Reference in New Issue
Block a user