/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | reshape_test.cc | 66 ComputeAndCompareLiteral(&builder, expected_literal, {input.get()}, in XLA_TEST_P() 80 ComputeAndCompareLiteral(&builder, expected_literal, {input.get()}, in XLA_TEST_P() 94 ComputeAndCompareLiteral(&builder, expected_literal, {input.get()}, in XLA_TEST_P() 113 ComputeAndCompareLiteral(&builder, expected_literal, {input.get()}, in XLA_TEST_P() 129 ComputeAndCompareLiteral(&builder, expected_literal, {input.get()}, in XLA_TEST_P() 143 ComputeAndCompareLiteral(&builder, expected_literal, {input.get()}, in XLA_TEST_P() 158 ComputeAndCompareLiteral(&builder, expected_literal, {input.get()}, in XLA_TEST_P() 172 ComputeAndCompareLiteral(&builder, expected_literal, {input.get()}, in XLA_TEST_P() 186 ComputeAndCompareLiteral(&builder, expected_literal, {input.get()}, in XLA_TEST_P() 200 ComputeAndCompareLiteral(&builder, expected_literal, {input.get()}, in XLA_TEST_P() [all …]
|
D | broadcast_simple_test.cc | 301 ComputeAndCompareLiteral(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 369 ComputeAndCompareLiteral( in XLA_TEST_P() 394 ComputeAndCompareLiteral(&b, expected, {r3.get(), r1.get()}, in XLA_TEST_F() 408 ComputeAndCompareLiteral(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 421 ComputeAndCompareLiteral(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 435 ComputeAndCompareLiteral(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 449 ComputeAndCompareLiteral(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 463 ComputeAndCompareLiteral(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 476 ComputeAndCompareLiteral(&b, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F() 616 ComputeAndCompareLiteral( in XLA_TEST_P() [all …]
|
D | reduce_window_test.cc | 133 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateR0<float>(43.0), {}, in XLA_TEST_P() 141 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateR1<float>({100, 1}), in XLA_TEST_P() 149 ComputeAndCompareLiteral(&builder_, in XLA_TEST_P() 159 ComputeAndCompareLiteral(&builder_, in XLA_TEST_P() 173 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {}, in XLA_TEST_P() 188 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {}, in XLA_TEST_P() 202 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {}, in XLA_TEST_P() 219 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*res), {}, in XLA_TEST_P() 241 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {}, in XLA_TEST_P() 278 ComputeAndCompareLiteral(&builder_, LiteralUtil::CreateFromArray(*result), {}, in XLA_TEST_P() [all …]
|
D | client_library_test_base.h | 185 void ComputeAndCompareLiteral(XlaBuilder* builder, const Literal& expected, 188 void ComputeAndCompareLiteral(XlaBuilder* builder, const Literal& expected, 453 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR0() 469 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR0() 478 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR1() 494 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR1() 504 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR2() 521 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR2() 531 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR3() 548 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR3() [all …]
|
D | unary_op_test.cc | 107 ComputeAndCompareLiteral(&builder, expected, {}, ErrorSpec(1e-6f)); in AbsTestHelper() 120 ComputeAndCompareLiteral(&builder, expected, {}, ErrorSpec(1e-6f)); in SignTestHelper() 133 ComputeAndCompareLiteral(&builder, expected, {}, ErrorSpec(1e-6f)); in SignAbsTestHelper() 177 ComputeAndCompareLiteral(&builder, expected, {}, ErrorSpec(1e-6f)); in XLA_TEST_F()
|
D | convolution_test.cc | 293 ComputeAndCompareLiteral(&builder, expected_r5, in XLA_TEST_F() 356 ComputeAndCompareLiteral(&builder, expected_r4, in RunTest() 420 ComputeAndCompareLiteral(&builder, expected_r4, in RunTest() 488 ComputeAndCompareLiteral(&builder, expected_r4, in RunTest() 553 ComputeAndCompareLiteral(&builder, expected_r4, in RunTest() 619 ComputeAndCompareLiteral(&builder, expected_r4_relaid, in RunTest() 689 ComputeAndCompareLiteral(&builder, expected_r4, in RunTest() 760 ComputeAndCompareLiteral(&builder, expected_r4_relaid, in RunTest() 835 ComputeAndCompareLiteral(&builder, expected_r4_relaid, in RunTest() 901 ComputeAndCompareLiteral(&builder, expected_r4, in RunTest() [all …]
|
D | matrix_ops_simple_test.cc | 70 this->ComputeAndCompareLiteral(&builder, expected, {}, ErrorSpec(1e-5)); in XLA_TYPED_TEST() 98 this->ComputeAndCompareLiteral(&builder, expected, {}, ErrorSpec(1e-5)); in XLA_TYPED_TEST() 117 this->ComputeAndCompareLiteral(&builder, expected, {}, ErrorSpec(1e-6)); in XLA_TYPED_TEST()
|
D | dynamic_ops_test.cc | 145 ComputeAndCompareLiteral(&builder, expected_values, {start_data.get()}); in RunR1() 180 ComputeAndCompareLiteral(&builder, expected_values, argument_ptrs); in RunR2() 214 ComputeAndCompareLiteral(&builder, expected_values, argument_ptrs); in RunR3() 398 ComputeAndCompareLiteral(&builder, expected_value, {}); in RunR0() 429 ComputeAndCompareLiteral(&builder, expected_values, {start_data.get()}); in RunR1() 468 ComputeAndCompareLiteral(&builder, expected_values, argument_ptrs); in RunR2() 508 ComputeAndCompareLiteral(&builder, expected_values, argument_ptrs); in RunR3()
|
D | reverse_test.cc | 105 ComputeAndCompareLiteral(&builder, expected, {}); in TEST_P()
|
D | client_library_test_base.cc | 166 ClientLibraryTestBase::ComputeAndCompareLiteral(builder, expected_literal, in ComputeAndCompareR1() 170 void ClientLibraryTestBase::ComputeAndCompareLiteral( in ComputeAndCompareLiteral() function in xla::ClientLibraryTestBase 177 void ClientLibraryTestBase::ComputeAndCompareLiteral( in ComputeAndCompareLiteral() function in xla::ClientLibraryTestBase
|
D | params_test.cc | 466 ComputeAndCompareLiteral(&builder, literal, {data.get()}, ErrorSpec(1e-3)); in XLA_TEST_F() 478 ComputeAndCompareLiteral(&builder, literal, {data.get()}, ErrorSpec(1e-3)); in XLA_TEST_F()
|
D | convolution_test_1d.cc | 128 ComputeAndCompareLiteral(&builder, expected_r3, in TestImpl()
|
D | slice_test.cc | 199 ComputeAndCompareLiteral(&builder, expected_literal, {}, ErrorSpec(0.000001), in XLA_TEST_F()
|
D | convolution_variants_test.cc | 1354 ComputeAndCompareLiteral(&builder, expected_literal, {}, error_spec_); in XLA_TEST_F() 1382 ComputeAndCompareLiteral(&builder, expected_literal, {}, error_spec_); in XLA_TEST_F()
|
D | dot_operation_test.cc | 87 ComputeAndCompareLiteral(&builder, in XLA_TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
D | infeed_test.cc | 49 ComputeAndCompareLiteral(&builder, literal, {}); in TestInfeedRoundTrip()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
D | cpu_infeed_test.cc | 55 ComputeAndCompareLiteral(&builder, literal, {}); in TestInfeedRoundTrip()
|
/external/tensorflow/tensorflow/compiler/xla/client/lib/ |
D | slicing_test.cc | 216 ComputeAndCompareLiteral(&builder, l, {input_data.get(), index_data.get()}); in XLA_TEST_F()
|
D | math_test.cc | 93 ComputeAndCompareLiteral(&b, expected, {}); in TestIsInfOrNan() 105 ComputeAndCompareLiteral( in TestIsNegZero()
|