Home
last modified time | relevance | path

Searched refs:StatelessRandomUniformIntOp (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dstateless_random_ops.cc189 class StatelessRandomUniformIntOp : public XlaOpKernel { class
191 explicit StatelessRandomUniformIntOp(OpKernelConstruction* ctx) in StatelessRandomUniformIntOp() function in tensorflow::__anon8e4597150411::StatelessRandomUniformIntOp
230 TF_DISALLOW_COPY_AND_ASSIGN(StatelessRandomUniformIntOp);
238 StatelessRandomUniformIntOp);
H A Dstateless_random_ops_v2.cc278 class StatelessRandomUniformIntOp : public XlaOpKernel { class
280 explicit StatelessRandomUniformIntOp(OpKernelConstruction* ctx) in StatelessRandomUniformIntOp() function in tensorflow::__anon6a88459c0211::StatelessRandomUniformIntOp
332 TF_DISALLOW_COPY_AND_ASSIGN(StatelessRandomUniformIntOp);
340 StatelessRandomUniformIntOp);
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstateless_random_ops.cc119 class StatelessRandomUniformIntOp : public StatelessRandomOpBase { class
250 StatelessRandomUniformIntOp<DEVICE##Device, TYPE>)
H A Dstateless_random_ops_v2.cc107 class StatelessRandomUniformIntOp : public StatelessRandomOpBase { class
289 StatelessRandomUniformIntOp<DEVICE##Device, TYPE>)
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf_with_tf2xla.cc249 TypeID::get<TF::StatelessRandomUniformIntOp>(), in IsOpAllowedTf2XlaFallback()