Searched refs:OneHotOp (Results 1 – 2 of 2) sorted by relevance
26 class OneHotOp : public XlaOpKernel { class28 explicit OneHotOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in OneHotOp() function in tensorflow::__anonc3bfa5090111::OneHotOp76 TF_DISALLOW_COPY_AND_ASSIGN(OneHotOp);79 REGISTER_XLA_OP(Name("OneHot").CompileTimeConstantInput("depth"), OneHotOp);
44 class OneHotOp : public OpKernel { class46 explicit OneHotOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in OneHotOp() function in tensorflow::OneHotOp125 TF_DISALLOW_COPY_AND_ASSIGN(OneHotOp);134 OneHotOp<CPUDevice, type, index_type>);178 OneHotOp<GPUDevice, type, index_type>);