Home
last modified time | relevance | path

Searched refs:RngOpInternal (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.h193 StatusOr<XlaOp> RngOpInternal(RandomDistribution distribution,
Dmlir_hlo_builder.cc330 StatusOr<XlaOp> MlirHloBuilder::RngOpInternal( in RngOpInternal() function in xla::MlirHloBuilder
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.cc2184 return RngOpInternal(distribution, parameters, shape); in RngOp()
2188 StatusOr<XlaOp> XlaBuilder::RngOpInternal(RandomDistribution distribution, in RngOpInternal() function in xla::XlaBuilder
Dxla_builder.h952 virtual StatusOr<XlaOp> RngOpInternal(RandomDistribution distribution,