| /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/tf2xla/kernels/ |
| D | scatter_nd_op.cc | 140 const std::function<xla::XlaOp(xla::XlaOp, xla::XlaOp, xla::XlaBuilder*)>& in CompileTensorScatter()
231 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 | 151 std::function<xla::XlaOp(const xla::XlaOp&, const xla::XlaOp&, in ResourceScatterOp()
|
| D | stateful_random_ops.cc | 175 std::function<SamplerReturnType(Algorithm, xla::XlaOp, xla::XlaOp, in CompileImpl()
|
| D | image_ops.cc | 39 const std::array<xla::XlaOp, 3>& rgb, in RGBToHSV()
72 const std::array<xla::XlaOp, 3>& hsv, in HSVToRGB()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | xla_builder.h | 49 class XlaOp; variable
73 XlaOp() : handle_(-1), builder_(nullptr) { in XlaOp() function
110 explicit XlaOp(XlaBuilder* builder) : handle_(-1), builder_(builder) {} in XlaOp() function
111 XlaOp(int64 handle, XlaBuilder* builder) in XlaOp() function
|
| D | xla_builder_test.cc | 97 ::testing::Matcher<const ::xla::HloInstruction*> matches_pattern) { in TEST_F()
128 ::testing::Matcher<const ::xla::HloInstruction*> matches_pattern) { in TEST_F()
|
| /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/lib/ |
| D | prng.cc | 131 XlaOp Uint32sToUint64(std::array<XlaOp, 2> u32s) { in Uint32sToUint64()
340 const std::array<XlaOp, 2>& u128, XlaOp u64, in Uint128AddUint64()
352 std::array<XlaOp, 2> Uint32sToUint128(const std::array<XlaOp, 4>& u32s) { in Uint32sToUint128()
357 std::array<XlaOp, 4> Uint128ToUint32s(const std::array<XlaOp, 2>& u128) { in Uint128ToUint32s()
369 XlaOp Uint128ToOp(std::array<XlaOp, 2> u128) { in Uint128ToOp()
|
| D | arithmetic_test.cc | 80 std::function<void(XlaOp, PrimitiveType)> MinMaxImpl) { in TestArgMinMaxImpl()
|
| D | slicing.cc | 227 const std::function<XlaOp(XlaOp, XlaOp)>& combiner) { in TorchScatterDense()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_expression.cc | 51 XlaExpression XlaExpression::XlaOp(xla::XlaOp value, DataType dtype) { in XlaOp() function in tensorflow::XlaExpression
|