Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Drandom_shuffle_op.cc63 class RandomShuffleOp : public OpKernel { class
65 explicit RandomShuffleOp(OpKernelConstruction* context) : OpKernel(context) { in RandomShuffleOp() function in tensorflow::RandomShuffleOp
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Drandom_ops.cc70 class RandomShuffleOp : public XlaOpKernel { class
72 explicit RandomShuffleOp(OpKernelConstruction* ctx) : XlaOpKernel(ctx) {} in RandomShuffleOp() function in tensorflow::__anone8f595e80111::RandomShuffleOp