| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | prng_test.cc | 54 RngUniform( in UniformTest()
152 RngUniform(ConstantR0<int32>(&builder, 0), in UniformChiSquared()
199 RngUniform(ConstantR0<float>(b.get(), 0), ConstantR0<float>(b.get(), 1), in XLA_TEST_F()
240 RngUniform(ConstantR0<float>(&builder, 0), ConstantR0<float>(&builder, 1), in XLA_TEST_F()
299 auto a = RngUniform(ConstantR0<int32>(&builder, 0), in XLA_TEST_F()
302 auto b = RngUniform(ConstantR0<int32>(&builder, 0), in XLA_TEST_F()
340 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 | 57 xla::XlaOp result = xla::RngUniform(XlaHelpers::Zero(b, dtype), in Compile()
143 xla::XlaOp keys = xla::RngUniform(zero, max_value, key_shape); in Compile()
162 xla::RngUniform(xla::ConstantR0<int32>(builder, 0), in Compile()
245 ctx->SetOutput(0, xla::RngUniform(minval, maxval, xla_shape)); in Compile()
306 auto uniform = xla::RngUniform(min_positive, one, xla_shape); in Compile()
339 xla::XlaOp uniform = xla::RngUniform(min_positive, one, xla_shape); in Compile()
|
| D | categorical_op.cc | 138 auto uniforms = xla::RngUniform( in GetLogUniforms()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | ops.cc | 249 ops.def("RngUniform", &RngUniform, py::arg("a"), py::arg("b"), in BuildOpsSubmodule()
|
| D | xla_client_test.py | 1225 ops.RngUniform(
1243 ops.RngUniform(
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 815 XlaOp RngUniform(XlaOp a, XlaOp b, const Shape& shape);
1373 friend XlaOp RngUniform(XlaOp a, XlaOp b, const Shape& shape);
2411 XlaOp RngUniform(XlaOp a, XlaOp b, const Shape& shape);
|
| D | xla_builder.cc | 2202 XlaOp XlaBuilder::RngUniform(XlaOp a, XlaOp b, const Shape& shape) { in RngUniform() function in xla::XlaBuilder
4838 XlaOp RngUniform(const XlaOp a, const XlaOp b, const Shape& shape) { in RngUniform() function
4839 return a.builder()->RngUniform(a, b, shape); in RngUniform()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 936 value_map[op] = xla::RngUniform(a, b, xla::TypeToShape(op.getType())); in ExportXlaOp()
|
| /external/tensorflow/tensorflow/compiler/xla/g3doc/ |
| D | operation_semantics.md | 2327 ## RngUniform section in Operation Semantics
2330 [`XlaBuilder::RngUniform`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_build…
2340 <b>`RngUniform(a, b, shape)`</b>
|