Searched refs:OnesLikeOp (Results 1 – 2 of 2) sorted by relevance
323 class OnesLikeOp : public OpKernel { class325 explicit OnesLikeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in OnesLikeOp() function in tensorflow::OnesLikeOp340 OnesLikeOp<dev##Device, type>)353 OnesLikeOp<CPUDevice, int32>);369 OnesLikeOp<CPUDevice, int32>);
235 class OnesLikeOp : public XlaOpKernel { class237 explicit OnesLikeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in OnesLikeOp() function in tensorflow::__anon812f91ee0111::OnesLikeOp247 REGISTER_XLA_OP(Name("OnesLike"), OnesLikeOp);