| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | client_library_test_base.cc | 60 ClientLibraryTestBase::ClientLibraryTestBase( in ClientLibraryTestBase() function in xla::ClientLibraryTestBase
83 ClientLibraryTestBase::ClientLibraryTestBase(se::Platform* platform) in ClientLibraryTestBase() function in xla::ClientLibraryTestBase
100 std::string ClientLibraryTestBase::TestName() const { in TestName()
104 StatusOr<std::unique_ptr<GlobalData>> ClientLibraryTestBase::Execute( in Execute()
111 StatusOr<Literal> ClientLibraryTestBase::ExecuteAndTransfer( in ExecuteAndTransfer()
123 StatusOr<Literal> ClientLibraryTestBase::ExecuteAndTransfer( in ExecuteAndTransfer()
131 StatusOr<Literal> ClientLibraryTestBase::ExecuteAndTransferReference( in ExecuteAndTransferReference()
144 std::string ClientLibraryTestBase::ExecuteToString( in ExecuteToString()
161 void ClientLibraryTestBase::ComputeAndCompareR1( in ComputeAndCompareR1()
165 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR1()
[all …]
|
| H A D | client_library_test_base.h | 64 class ClientLibraryTestBase : public ManifestCheckingTest {
66 explicit ClientLibraryTestBase(se::Platform* platform = nullptr);
69 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 …]
|
| H A D | iota_test.cc | 59 : public ClientLibraryTestBase,
87 class IotaR2Test : public ClientLibraryTestBase,
119 class IotaR3Test : public ClientLibraryTestBase,
|
| H A D | slice_test.cc | 38 class SliceTest : public ClientLibraryTestBase {};
211 class SliceR1Test : public ClientLibraryTestBase,
400 class SliceR2Test : public ClientLibraryTestBase,
491 class SliceR4Test : public ClientLibraryTestBase,
|
| H A D | reverse_test.cc | 74 class FloatReverseTest : public ClientLibraryTestBase,
112 class ReverseTest : public ClientLibraryTestBase {};
|
| H A D | bitcast_convert_test.cc | 35 class BitcastConvertTest : public ClientLibraryTestBase {
38 : ClientLibraryTestBase(platform) { in BitcastConvertTest()
|
| H A D | triangular_solve_test.cc | 39 using TriangularSolveTest = ClientLibraryTestBase;
40 using TriangularSolveLeftLookingTest = ClientLibraryTestBase;
445 : public ClientLibraryTestBase,
|
| H A D | matrix_ops_simple_test.cc | 48 class MatOpsSimpleTest : public ClientLibraryTestBase {};
178 : public ClientLibraryTestBase,
|
| H A D | cholesky_test.cc | 38 using CholeskyTest = ClientLibraryTestBase;
219 : public ClientLibraryTestBase,
|
| H A D | map_test.cc | 41 class MapTest : public ClientLibraryTestBase {
44 : ClientLibraryTestBase(platform) { in MapTest()
526 using MapTestWithFullOpt = ClientLibraryTestBase;
|
| H A D | vector_ops_simple_test.cc | 40 class VecOpsSimpleTest : public ClientLibraryTestBase {
43 : ClientLibraryTestBase(platform) { in VecOpsSimpleTest()
|
| H A D | reshape_motion_test.cc | 42 using ReshapeMotionTest = ClientLibraryTestBase;
|
| H A D | query_inferred_shape_test.cc | 30 class QueryInferredShapeTest : public ClientLibraryTestBase {};
|
| H A D | multidimensional_slice_test.cc | 32 class SliceTest : public ClientLibraryTestBase {};
|
| H A D | deep_graph_test.cc | 21 TEST_F(ClientLibraryTestBase, DeepGraph) { in TEST_F() argument
|
| H A D | log_test.cc | 29 class LogTest : public ClientLibraryTestBase {};
|
| H A D | bad_rng_shape_validation_test.cc | 34 class BadRngShapeValidationTest : public ClientLibraryTestBase {};
|
| H A D | execution_profile_test.cc | 26 class ExecutionProfileTest : public ClientLibraryTestBase {};
|
| H A D | axpy_simple_test.cc | 28 class AxpySimpleTest : public ClientLibraryTestBase {};
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| H A D | self_adjoint_eig_test.cc | 37 class SelfAdjointEigTest : public ClientLibraryTestBase {
40 ClientLibraryTestBase::SetUp(); in SetUp()
74 void TearDown() override { ClientLibraryTestBase::TearDown(); } in TearDown()
272 class RandomEighTest : public ClientLibraryTestBase,
|
| H A D | svd_test.cc | 40 class SVDTest : public ClientLibraryTestBase {
43 ClientLibraryTestBase::SetUp(); in SetUp()
62 void TearDown() override { ClientLibraryTestBase::TearDown(); } in TearDown()
|
| H A D | arithmetic_test.cc | 32 class ArithmeticTest : public ClientLibraryTestBase {
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| H A D | infeed_test.cc | 35 class InfeedTest : public ClientLibraryTestBase {
119 class BlockingInfeedTest : public ClientLibraryTestBase {};
163 class BlockingInfeedTest : public ClientLibraryTestBase {};
|
| H A D | bef_executable_test.cc | 25 XLA_TEST_F(ClientLibraryTestBase, GemmOnly) { in XLA_TEST_F() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | custom_call_test.cc | 56 class CustomCallTest : public ClientLibraryTestBase {};
219 public ClientLibraryTestBase {
|