| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | scatter.cc | 34 StatusOr<xla::XlaOp> XlaScatter( in XlaScatter()
37 const std::function<xla::XlaOp(xla::XlaOp, xla::XlaOp, xla::XlaBuilder*)>& in XlaScatter()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | scatter_nd_op.cc | 147 const std::function<xla::XlaOp(xla::XlaOp, xla::XlaOp, xla::XlaBuilder*)>& in CompileTensorScatter()
248 context, [](xla::XlaOp, xla::XlaOp y, xla::XlaBuilder*) { return y; }, in Compile()
|
| D | reduction_ops_common.cc | 41 xla::XlaBuilder* /*builder*/, const xla::XlaOp& /*input*/, in BuildFinalizer()
|
| D | variable_ops.cc | 167 std::function<xla::XlaOp(const xla::XlaOp&, const xla::XlaOp&, in ResourceScatterOp()
|
| D | stateful_random_ops.cc | 176 std::function<SamplerReturnType(Algorithm, xla::XlaOp, xla::XlaOp, in CompileImpl()
|
| D | image_ops.cc | 46 const std::array<xla::XlaOp, 3>& rgb, in RGBToHSV()
79 const std::array<xla::XlaOp, 3>& hsv, in HSVToRGB()
|
| /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 | 48 const std::function<XlaOp(const XlaOp&, const XlaOp&, in TestCompare()
59 const std::function<XlaOp(const XlaOp&, const XlaOp&, in TestMinMax()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 54 class XlaOp; variable
92 XlaOp() : handle_(-1), builder_(nullptr) { in XlaOp() function
129 explicit XlaOp(XlaBuilder* builder) : handle_(-1), builder_(builder) {} in XlaOp() function
130 XlaOp(int64_t handle, XlaBuilder* builder) in XlaOp() function
|
| D | xla_builder_test.cc | 98 ::testing::Matcher<const ::xla::HloInstruction*> matches_pattern) { in TEST_F()
129 ::testing::Matcher<const ::xla::HloInstruction*> matches_pattern) { in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | prng.cc | 129 XlaOp Uint32sToUint64(std::array<XlaOp, 2> u32s) { in Uint32sToUint64()
341 const std::array<XlaOp, 2>& u128, XlaOp u64, in Uint128AddUint64()
353 std::array<XlaOp, 2> Uint32sToUint128(const std::array<XlaOp, 4>& u32s) { in Uint32sToUint128()
358 std::array<XlaOp, 4> Uint128ToUint32s(const std::array<XlaOp, 2>& u128) { in Uint128ToUint32s()
370 XlaOp Uint128ToOp(std::array<XlaOp, 2> u128) { in Uint128ToOp()
|
| D | arithmetic_test.cc | 63 std::function<void(XlaOp, PrimitiveType)> MinMaxImpl) {} in TestArgMinMaxImpl()
|
| D | slicing.cc | 228 const std::function<XlaOp(XlaOp, XlaOp)>& combiner) { in TorchScatterDense()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_expression.cc | 52 XlaExpression XlaExpression::XlaOp(xla::XlaOp value, DataType dtype) { in XlaOp() function in tensorflow::XlaExpression
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | xla_client.py | 390 XlaOp = _xla.XlaOp variable
|