| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 256 xla::XlaOp result = xla::RngNormal(XlaHelpers::Zero(b, dtype), in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | prng_test.cc | 318 RngNormal(ConstantR0<float>(&builder, 0), ConstantR0<float>(&builder, 1), in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 678 LocalOp ComputationBuilder::RngNormal(const LocalOp& mu, const LocalOp& sigma, in RngNormal() function in xla::swig::ComputationBuilder
680 return xla::RngNormal(mu.op(), sigma.op(), shape); in RngNormal()
|
| D | local_computation_builder.h | 339 LocalOp RngNormal(const LocalOp& mu, const LocalOp& sigma,
|
| D | local_computation_builder.i | 325 %unignore xla::swig::ComputationBuilder::RngNormal;
|
| D | xla_client.py | 1517 def RngNormal(self, mu, sigma, dims): member in ComputationBuilder
1527 return self._client.RngNormal(mu, sigma, shape)
|
| D | xla_client_test.py | 1120 c.RngNormal(c.Constant(NumpyArrayF32(0.)), c.Constant(NumpyArrayF32(1.)),
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 524 XlaOp RngNormal(const XlaOp& mu, const XlaOp& sigma, const Shape& shape);
953 friend XlaOp RngNormal(const XlaOp& mu, const XlaOp& sigma,
1785 XlaOp RngNormal(const XlaOp& mu, const XlaOp& sigma, const Shape& shape);
|
| D | xla_builder.cc | 1809 XlaOp XlaBuilder::RngNormal(const XlaOp& mu, const XlaOp& sigma, in RngNormal() function in xla::XlaBuilder
3389 XlaOp RngNormal(const XlaOp& mu, const XlaOp& sigma, const Shape& shape) { in RngNormal() function
3390 return mu.builder()->RngNormal(mu, sigma, shape); in RngNormal()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2063 ## RngNormal section in Operation Semantics
2066 [`XlaBuilder::RngNormal`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_builde…
2073 <b>`RngNormal(mu, sigma, shape)`</b>
|