| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | client_library_test_base.cc | 61 ClientLibraryTestBase::ClientLibraryTestBase( in ClientLibraryTestBase() function in xla::ClientLibraryTestBase
84 ClientLibraryTestBase::ClientLibraryTestBase(se::Platform* platform) in ClientLibraryTestBase() function in xla::ClientLibraryTestBase
101 string ClientLibraryTestBase::TestName() const { in TestName()
105 StatusOr<std::unique_ptr<GlobalData>> ClientLibraryTestBase::Execute( in Execute()
112 StatusOr<Literal> ClientLibraryTestBase::ExecuteAndTransfer( in ExecuteAndTransfer()
124 StatusOr<Literal> ClientLibraryTestBase::ExecuteAndTransfer( in ExecuteAndTransfer()
132 StatusOr<Literal> ClientLibraryTestBase::ExecuteAndTransferReference( in ExecuteAndTransferReference()
145 string ClientLibraryTestBase::ExecuteToString( in ExecuteToString()
162 void ClientLibraryTestBase::ComputeAndCompareR1( in ComputeAndCompareR1()
166 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR1()
[all …]
|
| D | client_library_test_base.h | 66 class ClientLibraryTestBase : public ManifestCheckingTest {
68 explicit ClientLibraryTestBase(se::Platform* platform = nullptr);
71 ClientLibraryTestBase(se::Platform* platform,
449 void ClientLibraryTestBase::ComputeAndCompareR0( in ComputeAndCompareR0()
453 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR0()
458 void ClientLibraryTestBase::ComputeAndCompareR0( in ComputeAndCompareR0()
469 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR0()
474 void ClientLibraryTestBase::ComputeAndCompareR1( in ComputeAndCompareR1()
478 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR1()
483 void ClientLibraryTestBase::ComputeAndCompareR1( in ComputeAndCompareR1()
[all …]
|
| D | iota_test.cc | 59 : public ClientLibraryTestBase,
87 class IotaR2Test : public ClientLibraryTestBase,
119 class IotaR3Test : public ClientLibraryTestBase,
|
| D | reverse_test.cc | 75 class FloatReverseTest : public ClientLibraryTestBase,
113 class ReverseTest : public ClientLibraryTestBase {};
|
| D | bitcast_convert_test.cc | 35 class BitcastConvertTest : public ClientLibraryTestBase {
38 : ClientLibraryTestBase(platform) { in BitcastConvertTest()
|
| D | reshape_motion_test.cc | 43 using ReshapeMotionTest = ClientLibraryTestBase;
|
| D | query_inferred_shape_test.cc | 30 class QueryInferredShapeTest : public ClientLibraryTestBase {};
|
| D | multidimensional_slice_test.cc | 32 class SliceTest : public ClientLibraryTestBase {};
|
| D | deep_graph_test.cc | 21 TEST_F(ClientLibraryTestBase, DeepGraph) { in TEST_F() argument
|
| D | slice_test.cc | 39 class SliceTest : public ClientLibraryTestBase {};
212 class SliceR1Test : public ClientLibraryTestBase,
398 class SliceR2Test : public ClientLibraryTestBase,
489 class SliceR4Test : public ClientLibraryTestBase,
|
| D | log_test.cc | 29 class LogTest : public ClientLibraryTestBase {};
|
| D | bad_rng_shape_validation_test.cc | 34 class BadRngShapeValidationTest : public ClientLibraryTestBase {};
|
| D | execution_profile_test.cc | 26 class ExecutionProfileTest : public ClientLibraryTestBase {};
|
| D | axpy_simple_test.cc | 28 class AxpySimpleTest : public ClientLibraryTestBase {};
|
| D | matrix_ops_simple_test.cc | 50 class MatOpsSimpleTest : public ClientLibraryTestBase {};
180 : public ClientLibraryTestBase,
|
| D | cholesky_test.cc | 38 using CholeskyTest = ClientLibraryTestBase;
219 : public ClientLibraryTestBase,
|
| D | triangular_solve_test.cc | 38 using TriangularSolveTest = ClientLibraryTestBase;
39 using TriangularSolveLeftLookingTest = ClientLibraryTestBase;
452 : public ClientLibraryTestBase,
|
| D | floor_ceil_test.cc | 32 class FloorCeilTest : public ClientLibraryTestBase {
|
| D | fmax_fmin_test.cc | 28 class FmaxSimpleTest : public ClientLibraryTestBase {};
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd_test.cc | 40 class SVDTest : public ClientLibraryTestBase {
43 ClientLibraryTestBase::SetUp(); in SetUp()
62 void TearDown() override { ClientLibraryTestBase::TearDown(); } in TearDown()
|
| D | self_adjoint_eig_test.cc | 35 class SelfAdjointEigTest : public ClientLibraryTestBase {
38 ClientLibraryTestBase::SetUp(); in SetUp()
72 void TearDown() override { ClientLibraryTestBase::TearDown(); } in TearDown()
|
| D | logdet_test.cc | 32 using LogDetTest = xla::ClientLibraryTestBase;
|
| D | qr_test.cc | 34 using QrTest = xla::ClientLibraryTestBase;
|
| D | arithmetic_test.cc | 32 class ArithmeticTest : public ClientLibraryTestBase {
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | infeed_test.cc | 35 class InfeedTest : public ClientLibraryTestBase {
|