Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconstant_op.cc323 class OnesLikeOp : public OpKernel { class
325 explicit OnesLikeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in OnesLikeOp() function in tensorflow::OnesLikeOp
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dshape_op.cc235 class OnesLikeOp : public XlaOpKernel { class
237 explicit OnesLikeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in OnesLikeOp() function in tensorflow::__anonea35bc650111::OnesLikeOp