* Also fix unit test
* Implement InferShape and register them, give a stub Kernel method by LOG(INFO)