| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | prng_test.cc | 54 RngUniform( in UniformTest()
145 RngUniform(ConstantR0<int32>(&builder, 0), in UniformChiSquared()
190 RngUniform(ConstantR0<float>(b.get(), 0), ConstantR0<float>(b.get(), 1), in XLA_TEST_F()
230 RngUniform(ConstantR0<float>(&builder, 0), ConstantR0<float>(&builder, 1), in XLA_TEST_F()
289 auto a = RngUniform(ConstantR0<int32>(&builder, 0), in XLA_TEST_F()
292 auto b = RngUniform(ConstantR0<int32>(&builder, 0), in XLA_TEST_F()
330 RngUniform(ConstantR0<int32>(&builder, 0), in XLA_TEST_F()
|
| D | bad_rng_shape_validation_test.cc | 41 RngUniform(zero, one, default_constructed); in TEST_F()
58 RngUniform(zero, one, sans_layout); in TEST_F()
|
| D | compute_constant_test.cc | 130 RngUniform(ConstantR0<float>(&b, 1.1f), ConstantR0<float>(&b, 2.1f), in TEST_F()
|
| D | while_test.cc | 923 RngUniform(ConstantR0<int32>(&builder, 0), in XLA_TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | random_ops.cc | 51 xla::XlaOp result = xla::RngUniform(XlaHelpers::Zero(b, dtype), in Compile()
137 xla::XlaOp keys = xla::RngUniform(zero, max_value, key_shape); in Compile()
151 xla::RngUniform(xla::ConstantR0<int32>(builder, 0), in Compile()
230 ctx->SetOutput(0, xla::RngUniform(minval, maxval, xla_shape)); in Compile()
286 auto uniform = xla::RngUniform(min_positive, one, xla_shape); in Compile()
|
| D | categorical_op.cc | 119 auto uniforms = xla::RngUniform( in GetLogUniforms()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_computation_builder.cc | 683 LocalOp ComputationBuilder::RngUniform(const LocalOp& a, const LocalOp& b, in RngUniform() function in xla::swig::ComputationBuilder
685 return xla::RngUniform(a.op(), b.op(), shape); in RngUniform()
|
| D | local_computation_builder.h | 342 LocalOp RngUniform(const LocalOp& a, const LocalOp& b, const Shape& shape);
|
| D | local_computation_builder.i | 326 %unignore xla::swig::ComputationBuilder::RngUniform;
|
| D | xla_client.py | 1529 def RngUniform(self, a, b, dims): member in ComputationBuilder
1544 return self._client.RngUniform(a, b, shape)
|
| D | xla_client_test.py | 1131 c.RngUniform(c.Constant(NumpyArrayF32(lo)), c.Constant(NumpyArrayF32(hi)),
1144 c.RngUniform(c.Constant(NumpyArrayS32(lo)), c.Constant(NumpyArrayS32(hi)),
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 526 XlaOp RngUniform(const XlaOp& a, const XlaOp& b, const Shape& shape);
955 friend XlaOp RngUniform(const XlaOp& a, const XlaOp& b, const Shape& shape);
1789 XlaOp RngUniform(const XlaOp& a, const XlaOp& b, const Shape& shape);
|
| D | xla_builder.cc | 1814 XlaOp XlaBuilder::RngUniform(const XlaOp& a, const XlaOp& b, in RngUniform() function in xla::XlaBuilder
3393 XlaOp RngUniform(const XlaOp& a, const XlaOp& b, const Shape& shape) { in RngUniform() function
3394 return a.builder()->RngUniform(a, b, shape); in RngUniform()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2083 ## RngUniform section in Operation Semantics
2086 [`XlaBuilder::RngUniform`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_build…
2096 <b>`RngUniform(a, b, shape)`</b>
|