Home
last modified time | relevance | path

Searched refs:RngBitGenerator (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dstateless_random_ops.cc49 xla::XlaOp result = xla::RngBitGenerator(xla::RandomAlgorithm::RNG_PHILOX, in GetBitGeneratorForDevice()
58 xla::RngBitGenerator(xla::RandomAlgorithm::RNG_DEFAULT, state, shape); in GetBitGeneratorForDevice()
Dstateful_random_ops.cc47 xla::RngBitGenerator(xla::RandomAlgorithm::RNG_PHILOX, state, shape); in BitGen()
56 xla::XlaOp result = xla::RngBitGenerator( in BitGen()
Dstateless_random_ops_v2.cc75 xla::XlaOp result = xla::RngBitGenerator(alg, state, shape); in BitGenerator()
/external/tensorflow/tensorflow/compiler/xla/python/
Dops.cc245 ops.def("RngBitGenerator", &RngBitGenerator, py::arg("algorithm"), in BuildOpsSubmodule()
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h817 XlaOp RngBitGenerator(RandomAlgorithm algorithm, XlaOp initial_state,
1374 friend XlaOp RngBitGenerator(RandomAlgorithm algorithm, XlaOp initial_state,
2415 XlaOp RngBitGenerator(RandomAlgorithm algorithm, XlaOp initial_state,
Dxla_builder.cc2206 XlaOp XlaBuilder::RngBitGenerator(RandomAlgorithm algorithm, in RngBitGenerator() function in xla::XlaBuilder
4842 XlaOp RngBitGenerator(RandomAlgorithm algorithm, const XlaOp initial_state, in RngBitGenerator() function
4844 return initial_state.builder()->RngBitGenerator(algorithm, initial_state, in RngBitGenerator()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_parser_test.cc1136 "RngBitGenerator", in CreateTestCases()
1139 ENTRY %RngBitGenerator (p0: u64[2]) -> (u64[2], u32[11,17]) { in CreateTestCases()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc913 auto xla_result = xla::RngBitGenerator( in ExportXlaOp()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2350 ## RngBitGenerator section in Operation Semantics
2363 <b>`RngBitGenerator(algorithm, key, shape)`</b>