Home
last modified time | relevance | path

Searched refs:RngNormal (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dprng_test.cc328 RngNormal(ConstantR0<float>(&builder, 0), ConstantR0<float>(&builder, 1), in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Drandom_ops.cc271 xla::XlaOp result = xla::RngNormal(XlaHelpers::Zero(b, dtype), in Compile()
/external/tensorflow/tensorflow/compiler/xla/python/
Dops.cc247 ops.def("RngNormal", &RngNormal, py::arg("mu"), py::arg("sigma"), in BuildOpsSubmodule()
Dxla_client_test.py1209 ops.RngNormal(
/external/tensorflow/tensorflow/compiler/xla/client/
Dxla_builder.h813 XlaOp RngNormal(XlaOp mu, XlaOp sigma, const Shape& shape);
1372 friend XlaOp RngNormal(XlaOp mu, XlaOp sigma, const Shape& shape);
2407 XlaOp RngNormal(XlaOp mu, XlaOp sigma, const Shape& shape);
Dxla_builder.cc2198 XlaOp XlaBuilder::RngNormal(XlaOp mu, XlaOp sigma, const Shape& shape) { in RngNormal() function in xla::XlaBuilder
4834 XlaOp RngNormal(const XlaOp mu, const XlaOp sigma, const Shape& shape) { in RngNormal() function
4835 return mu.builder()->RngNormal(mu, sigma, shape); in RngNormal()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dmlir_hlo_to_hlo.cc926 value_map[op] = xla::RngNormal(mu, sigma, xla::TypeToShape(op.getType())); in ExportXlaOp()
/external/tensorflow/tensorflow/compiler/xla/g3doc/
Doperation_semantics.md2307 ## RngNormal section in Operation Semantics
2310 [`XlaBuilder::RngNormal`](https://www.tensorflow.org/code/tensorflow/compiler/xla/client/xla_builde…
2317 <b>`RngNormal(mu, sigma, shape)`</b>