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("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
|
} // end namespace tensorflow
|
||||||
|
Loading…
x
Reference in New Issue
Block a user