Searched refs:RngBitGenerator (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
D | stateless_random_ops.cc | 49 xla::XlaOp result = xla::RngBitGenerator(xla::RandomAlgorithm::RNG_PHILOX, in GetBitGeneratorForDevice() 58 xla::RngBitGenerator(xla::RandomAlgorithm::RNG_DEFAULT, state, shape); in GetBitGeneratorForDevice()
|
D | stateful_random_ops.cc | 47 xla::RngBitGenerator(xla::RandomAlgorithm::RNG_PHILOX, state, shape); in BitGen() 56 xla::XlaOp result = xla::RngBitGenerator( in BitGen()
|
D | stateless_random_ops_v2.cc | 75 xla::XlaOp result = xla::RngBitGenerator(alg, state, shape); in BitGenerator()
|
/external/tensorflow/tensorflow/compiler/xla/python/ |
D | ops.cc | 245 ops.def("RngBitGenerator", &RngBitGenerator, py::arg("algorithm"), in BuildOpsSubmodule()
|
/external/tensorflow/tensorflow/compiler/xla/client/ |
D | xla_builder.h | 817 XlaOp RngBitGenerator(RandomAlgorithm algorithm, XlaOp initial_state, 1374 friend XlaOp RngBitGenerator(RandomAlgorithm algorithm, XlaOp initial_state, 2415 XlaOp RngBitGenerator(RandomAlgorithm algorithm, XlaOp initial_state,
|
D | xla_builder.cc | 2206 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/ |
D | hlo_parser_test.cc | 1136 "RngBitGenerator", in CreateTestCases() 1139 ENTRY %RngBitGenerator (p0: u64[2]) -> (u64[2], u32[11,17]) { in CreateTestCases()
|
/external/tensorflow/tensorflow/compiler/mlir/xla/ |
D | mlir_hlo_to_hlo.cc | 913 auto xla_result = xla::RngBitGenerator( in ExportXlaOp()
|
/external/tensorflow/tensorflow/compiler/xla/g3doc/ |
D | operation_semantics.md | 2350 ## RngBitGenerator section in Operation Semantics 2363 <b>`RngBitGenerator(algorithm, key, shape)`</b>
|