Home
last modified time | relevance | path

Searched refs:StatelessRandomUniformFullIntOp (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dstateless_random_ops.cc240 class StatelessRandomUniformFullIntOp : public XlaOpKernel { class
242 explicit StatelessRandomUniformFullIntOp(OpKernelConstruction* ctx) in StatelessRandomUniformFullIntOp() function in tensorflow::__anon8e4597150411::StatelessRandomUniformFullIntOp
271 TF_DISALLOW_COPY_AND_ASSIGN(StatelessRandomUniformFullIntOp);
279 StatelessRandomUniformFullIntOp);
H A Dstateless_random_ops_v2.cc342 class StatelessRandomUniformFullIntOp : public XlaOpKernel { class
344 explicit StatelessRandomUniformFullIntOp(OpKernelConstruction* ctx) in StatelessRandomUniformFullIntOp() function in tensorflow::__anon6a88459c0211::StatelessRandomUniformFullIntOp
381 TF_DISALLOW_COPY_AND_ASSIGN(StatelessRandomUniformFullIntOp);
389 StatelessRandomUniformFullIntOp);
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dstateless_random_ops.cc156 class StatelessRandomUniformFullIntOp : public StatelessRandomOpBase { class
239 StatelessRandomUniformFullIntOp<DEVICE##Device, TYPE>)
H A Dstateless_random_ops_v2.cc151 class StatelessRandomUniformFullIntOp : public StatelessRandomOpBase { class
278 StatelessRandomUniformFullIntOp<DEVICE##Device, TYPE>)
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/expansions/
H A Drandom_op_spmd_expander.cc318 if (llvm::isa<mlir::TF::StatelessRandomUniformFullIntOp>(op)) { in ExpandOp()
320 mlir::TF::StatelessRandomUniformFullIntOp>(*layout, op); in ExpandOp()
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dspmd_expanders.cc311 TF::StatelessRandomUniformFullIntOp, RandomOpSPMDExpander);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf_with_tf2xla.cc246 TypeID::get<TF::StatelessRandomUniformFullIntOp>(), in IsOpAllowedTf2XlaFallback()