| /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 | 65 class ClientLibraryTestBase : public ::testing::Test {
67 explicit ClientLibraryTestBase(se::Platform* platform = nullptr);
70 ClientLibraryTestBase(se::Platform* platform,
418 void ClientLibraryTestBase::ComputeAndCompareR0( in ComputeAndCompareR0()
422 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR0()
427 void ClientLibraryTestBase::ComputeAndCompareR0( in ComputeAndCompareR0()
438 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR0()
443 void ClientLibraryTestBase::ComputeAndCompareR1( in ComputeAndCompareR1()
447 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR1()
452 void ClientLibraryTestBase::ComputeAndCompareR1( in ComputeAndCompareR1()
[all …]
|
| D | iota_test.cc | 34 : public ClientLibraryTestBase,
62 class IotaR2Test : public ClientLibraryTestBase,
89 class IotaR3Test : public ClientLibraryTestBase,
116 class IotaR3PredTest : public ClientLibraryTestBase,
|
| D | reverse_test.cc | 74 class FloatReverseTest : public ClientLibraryTestBase,
112 class ReverseTest : public ClientLibraryTestBase {};
|
| D | bitcast_convert_test.cc | 35 class BitcastConvertTest : public ClientLibraryTestBase {
38 : ClientLibraryTestBase(platform) { in BitcastConvertTest()
|
| D | cholesky_test.cc | 37 using CholeskyTest = ClientLibraryTestBase;
156 : public ClientLibraryTestBase,
|
| D | slice_test.cc | 39 class SliceTest : public ClientLibraryTestBase {};
212 class SliceR1Test : public ClientLibraryTestBase,
387 class SliceR2Test : public ClientLibraryTestBase,
478 class SliceR4Test : public ClientLibraryTestBase,
|
| 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 | log_test.cc | 29 class LogTest : public ClientLibraryTestBase {};
|
| D | triangular_solve_test.cc | 36 using TriangularSolveTest = ClientLibraryTestBase;
37 using TriangularSolveLeftLookingTest = ClientLibraryTestBase;
442 : public ClientLibraryTestBase,
|
| D | matrix_ops_simple_test.cc | 50 class MatOpsSimpleTest : public ClientLibraryTestBase {};
180 : public ClientLibraryTestBase,
|
| D | reduce_precision_test.cc | 44 class ReducePrecisionAccuracyTest : public ClientLibraryTestBase,
248 class ReducePrecisionInsertionTest : 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 | floor_ceil_test.cc | 32 class FloorCeilTest : public ClientLibraryTestBase {
|
| D | fmax_fmin_test.cc | 27 class FmaxSimpleTest : public ClientLibraryTestBase {};
|
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | svd_test.cc | 38 class SVDTest : public ClientLibraryTestBase {
41 ClientLibraryTestBase::SetUp(); in SetUp()
57 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 | qr_test.cc | 33 using QrTest = xla::ClientLibraryTestBase;
|
| D | arithmetic_test.cc | 28 using ArithmeticTest = ClientLibraryTestBase;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | infeed_test.cc | 35 class InfeedTest : public ClientLibraryTestBase {
|