Home
last modified time | relevance | path

Searched refs:ComputeAndCompareLiteral (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dreshape_test.cc66 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 …]
Dbroadcast_simple_test.cc301 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 …]
Dreduce_window_test.cc133 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 …]
Dclient_library_test_base.h185 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 …]
Dunary_op_test.cc107 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()
Dconvolution_test.cc293 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 …]
Dmatrix_ops_simple_test.cc70 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()
Ddynamic_ops_test.cc145 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()
Dreverse_test.cc105 ComputeAndCompareLiteral(&builder, expected, {}); in TEST_P()
Dclient_library_test_base.cc166 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
Dparams_test.cc466 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()
Dconvolution_test_1d.cc128 ComputeAndCompareLiteral(&builder, expected_r3, in TestImpl()
Dslice_test.cc199 ComputeAndCompareLiteral(&builder, expected_literal, {}, ErrorSpec(0.000001), in XLA_TEST_F()
Dconvolution_variants_test.cc1354 ComputeAndCompareLiteral(&builder, expected_literal, {}, error_spec_); in XLA_TEST_F()
1382 ComputeAndCompareLiteral(&builder, expected_literal, {}, error_spec_); in XLA_TEST_F()
Ddot_operation_test.cc87 ComputeAndCompareLiteral(&builder, in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dinfeed_test.cc49 ComputeAndCompareLiteral(&builder, literal, {}); in TestInfeedRoundTrip()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_infeed_test.cc55 ComputeAndCompareLiteral(&builder, literal, {}); in TestInfeedRoundTrip()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dslicing_test.cc216 ComputeAndCompareLiteral(&builder, l, {input_data.get(), index_data.get()}); in XLA_TEST_F()
Dmath_test.cc93 ComputeAndCompareLiteral(&b, expected, {}); in TestIsInfOrNan()
105 ComputeAndCompareLiteral( in TestIsNegZero()