Searched refs:FakeParamOp (Results 1 – 1 of 1) sorted by relevance
631 class FakeParamOp : public OpKernel { class633 explicit FakeParamOp(OpKernelConstruction* context) : OpKernel(context) { in FakeParamOp() function in tensorflow::__anonf4cf7ed10111::FakeParamOp663 REGISTER_KERNEL_BUILDER(Name("FakeParam").Device(DEVICE_CPU), FakeParamOp);664 REGISTER_KERNEL_BUILDER(Name("FakeParam").Device(DEVICE_GPU), FakeParamOp);