Searched refs:OnesLikeOp (Results 1 - 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ | ||
H A D | shape_op.cc | 231 class OnesLikeOp : public XlaOpKernel { class in namespace:tensorflow::__anon25455 233 explicit OnesLikeOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} function in class:tensorflow::__anon25455::OnesLikeOp 243 REGISTER_XLA_OP(Name("OnesLike"), OnesLikeOp); member in namespace:tensorflow::__anon25455 |
/external/tensorflow/tensorflow/core/kernels/ | ||
H A D | constant_op.cc | 344 class OnesLikeOp : public OpKernel { class in namespace:tensorflow 346 explicit OnesLikeOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} function in class:tensorflow::OnesLikeOp 361 OnesLikeOp<dev##Device, type>) 374 OnesLikeOp<CPUDevice, int32>); member in namespace:tensorflow 390 OnesLikeOp<CPUDevice, int32>); member in namespace:tensorflow |
Completed in 281 milliseconds