Searched defs:ComputationDataHandle (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/lib/ |
| D | scatter.cc | 33 xla::StatusOr<xla::ComputationDataHandle> XlaScatter( in XlaScatter()
|
| D | triangular_solve.cc | 483 auto prepend_zeros = [&](std::array<xla::ComputationDataHandle, 2> starts) { in TriangularSolveLeftLooking()
501 std::array<xla::ComputationDataHandle, 2> starts) { in TriangularSolveLeftLooking()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | pred_test.cc | 32 void TestCompare(bool lhs, bool rhs, bool expected, in TestCompare()
34 const ComputationDataHandle&, in TestCompare()
|
| D | scalar_computations_test.cc | 46 void TestCompare(NativeT lhs, NativeT rhs, bool expected, in TestCompare()
48 const ComputationDataHandle&, in TestCompare()
59 void TestMinMax(NativeT lhs, NativeT rhs, NativeT expected, in TestMinMax()
61 const ComputationDataHandle&, in TestMinMax()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | image_ops.cc | 28 const std::array<xla::ComputationDataHandle, 3>& rgb, DataType dtype, in RGBToHSV()
59 const std::array<xla::ComputationDataHandle, 3>& hsv, DataType dtype) { in HSVToRGB()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | xla_data.proto | 286 message ComputationDataHandle { message
|