Home
last modified time | relevance | path

Searched defs:OneHotOp (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Done_hot_op.cc26 class OneHotOp : public XlaOpKernel { class
28 explicit OneHotOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) { in OneHotOp() function in tensorflow::__anon2cdc51a70111::OneHotOp
/external/tensorflow/tensorflow/core/kernels/
Done_hot_op.cc45 class OneHotOp : public OpKernel { class
47 explicit OneHotOp(OpKernelConstruction* ctx) : OpKernel(ctx) { in OneHotOp() function in tensorflow::OneHotOp