Searched defs:XlaOp (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | scatter.cc | 33 xla::StatusOr<xla::XlaOp> XlaScatter( in XlaScatter()
36 const std::function<xla::XlaOp(xla::XlaOp, xla::XlaOp, xla::XlaBuilder*)>& in XlaScatter()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 115 XlaOp StatelessRngUniformU32(std::array<XlaOp, 2> key, const Shape& shape) { in StatelessRngUniformU32()
144 XlaOp StatelessRngUniformU64(std::array<XlaOp, 2> key, const Shape& shape) { in StatelessRngUniformU64()
186 XlaOp StatelessRngUniform(std::array<XlaOp, 2> seeds, const Shape& shape, in StatelessRngUniform()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | pred_test.cc | 33 std::function<XlaOp(const xla::XlaOp&, const xla::XlaOp&, in TestCompare()
|
| D | scalar_computations_test.cc | 49 const std::function<XlaOp(const XlaOp&, const XlaOp&, in TestCompare()
60 const std::function<XlaOp(const XlaOp&, const XlaOp&, in TestMinMax()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 54 XlaOp() : handle_(-1), builder_(nullptr) { in XlaOp() function
91 explicit XlaOp(XlaBuilder* builder) : handle_(-1), builder_(builder) {} in XlaOp() function
92 XlaOp(int64 handle, XlaBuilder* builder) in XlaOp() function
|
| D | xla_builder_test.cc | 95 ::testing::Matcher<const ::xla::HloInstruction*> matches_pattern) { in TEST_F()
126 ::testing::Matcher<const ::xla::HloInstruction*> matches_pattern) { in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_expression.cc | 41 XlaExpression XlaExpression::XlaOp(xla::XlaOp value, DataType dtype) { in XlaOp() function in tensorflow::XlaExpression
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reduction_ops_common.cc | 41 xla::XlaBuilder* /*builder*/, const xla::XlaOp& /*input*/, in BuildFinalizer()
|
| D | stateful_random_ops.cc | 184 std::function<sampler_return_type(xla::XlaOp, xla::XlaOp, in CompileImpl()
|
| D | variable_ops.cc | 153 std::function<xla::XlaOp(const xla::XlaOp&, const xla::XlaOp&, in ResourceScatterOp()
|
| D | image_ops.cc | 37 const std::array<xla::XlaOp, 3>& rgb, in RGBToHSV()
70 const std::array<xla::XlaOp, 3>& hsv, in HSVToRGB()
|