Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dstateful_random_ops.cc262 class RngSkipOp : public OpKernel { class
264 explicit RngSkipOp(OpKernelConstruction* ctx) : OpKernel(ctx) {} in RngSkipOp() function in tensorflow::RngSkipOp
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstateful_random_ops.cc428 class RngSkipOp : public XlaOpKernel { class
430 explicit RngSkipOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in RngSkipOp() function in tensorflow::__anon8ba4988f0111::RngSkipOp