Home
last modified time | relevance | path

Searched refs:ComputeAndCompareR0 (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dconstants_test.cc34 ComputeAndCompareR0<int32>(&builder, 4, {}); in XLA_TEST_F()
48 ComputeAndCompareR0<float>(&builder, -7, {}); in XLA_TEST_F()
50 ComputeAndCompareR0<float>(&builder, 0.5, {}); in XLA_TEST_F()
56 ComputeAndCompareR0<int32>(&builder, -3, {}); in XLA_TEST_F()
62 ComputeAndCompareR0<float>(&builder, -3.2, {}); in XLA_TEST_F()
68 ComputeAndCompareR0<int32>(&builder, 0, {}); in XLA_TEST_F()
74 ComputeAndCompareR0<float>(&builder, 0.0, {}); in XLA_TEST_F()
92 ComputeAndCompareR0<int32>(&builder, 1, {}); in XLA_TEST_F()
98 ComputeAndCompareR0<float>(&builder, 1., {}); in XLA_TEST_F()
104 ComputeAndCompareR0<float>(&builder, std::numeric_limits<float>::epsilon(), in XLA_TEST_F()
[all …]
Dsvd_test.cc201 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()}, in XLA_TEST_F()
213 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()}, in XLA_TEST_F()
225 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()}, in XLA_TEST_F()
239 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()}, in XLA_TEST_F()
253 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()}, in XLA_TEST_F()
267 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()}, in XLA_TEST_F()
Dself_adjoint_eig_test.cc236 ComputeAndCompareR0<float>(&builder, expected_vals, {a_data.get()}, in XLA_TEST_F()
259 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()}, in XLA_TEST_F()
272 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()}, in XLA_TEST_F()
285 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()}, in XLA_TEST_F()
298 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()}, in XLA_TEST_F()
311 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()}, in XLA_TEST_F()
Dlogdet_test.cc50 ComputeAndCompareR0<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
70 ComputeAndCompareR0<float>(&builder, expected, {a_data.get()}, in XLA_TEST_F()
Dmath_test.cc236 ComputeAndCompareR0<float>(&builder, 0.0f, {zero_data.get()}, error_spec_); in XLA_TEST_F()
249 ComputeAndCompareR0<double>(&builder, 0.0f, {zero_data.get()}, error_spec_); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dscalar_computations_test.cc55 ComputeAndCompareR0<bool>(&builder, expected, {}); in TestCompare()
66 ComputeAndCompareR0<NativeT>(&builder, expected, {}); in TestMinMax()
74 ComputeAndCompareR0<float>(&builder, 2.1f, {}, error_spec_); in XLA_TEST_F()
81 ComputeAndCompareR0<float>(&builder, -2.1f, {}, error_spec_); in XLA_TEST_F()
88 ComputeAndCompareR0<int32>(&builder, -2, {}); in XLA_TEST_F()
95 ComputeAndCompareR0<float>(&builder, 7.6f, {}, error_spec_); in XLA_TEST_F()
102 ComputeAndCompareR0<int32>(&builder, 7, {}); in XLA_TEST_F()
109 ComputeAndCompareR0<uint32>(&builder, 92, {}); in XLA_TEST_F()
116 ComputeAndCompareR0<uint8>(&builder, 92, {}); in XLA_TEST_F()
125 ComputeAndCompareR0<uint64>(&builder, a + b, {}); in XLA_TEST_F()
[all …]
Dpred_test.cc40 ComputeAndCompareR0<bool>(&builder, expected, {}); in TestCompare()
47 ComputeAndCompareR0<bool>(&builder, true, {}); in TEST_F()
53 ComputeAndCompareR0<bool>(&builder, false, {}); in TEST_F()
100 ComputeAndCompareR0<bool>(&builder, true, {}); in TEST_F()
107 ComputeAndCompareR0<bool>(&builder, false, {}); in TEST_F()
114 ComputeAndCompareR0<bool>(&builder, false, {}); in TEST_F()
125 ComputeAndCompareR0<bool>(&builder, true, {}); in TEST_F()
136 ComputeAndCompareR0<bool>(&builder, false, {}); in TEST_F()
Dvector_ops_reduce_test.cc64 ComputeAndCompareR0<float>(&builder_, -4.2f, {}, errspec_); in TEST_F()
78 ComputeAndCompareR0<float>(&builder_, expected, {}, errspec_); in TEST_F()
89 ComputeAndCompareR0<float>(&builder_, 2.6f, {}, errspec_); in TEST_F()
100 ComputeAndCompareR0<float>(&builder_, 4.0f, {}, errspec_); in TEST_F()
200 ComputeAndCompareR0<float>(&builder_, 63.0, {}, errspec_); in TEST_F()
Dconditional_test.cc188 ComputeAndCompareR0<float>(&builder, 56.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
216 ComputeAndCompareR0<float>(&builder, expected, {branch_index_arg.get()}, in XLA_TEST_P()
231 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
268 ComputeAndCompareR0<float>(&builder, expected, {branch_index_arg.get()}, in XLA_TEST_P()
284 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
297 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
311 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
324 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
338 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
360 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_); in XLA_TEST_F()
[all …]
Dbfloat16_test.cc58 ComputeAndCompareR0<bfloat16>(&builder, static_cast<bfloat16>(3.0f), {}, in XLA_TEST_F()
67 ComputeAndCompareR0<bfloat16>(&builder, static_cast<bfloat16>(1.387f), {}, in XLA_TEST_F()
75 ComputeAndCompareR0<bfloat16>(&builder, static_cast<bfloat16>(-2.1f), {}, in XLA_TEST_F()
Dreduce_test.cc116 ComputeAndCompareR0<float>(&builder, expected, {input_global_data.get()}, in RunR1ToR0Test()
151 ComputeAndCompareR0<bool>(&builder, expected, {input_global_data.get()}); in RunR1ToR0PredTest()
225 ComputeAndCompareR0<float>(&builder, expected, {input_global_data.get()}, in RunR2ToR0Test()
570 ComputeAndCompareR0<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
582 ComputeAndCompareR0<float>(&builder, expected, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
597 ComputeAndCompareR0<float>(&builder, input_max, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
613 ComputeAndCompareR0<float>(&builder, input_min, {}, ErrorSpec(0.0001)); in XLA_TEST_F()
625 ComputeAndCompareR0<uint32>(&builder, 1, {}); in XLA_TEST_F()
637 ComputeAndCompareR0<uint32>(&builder, 2, {}); in XLA_TEST_F()
658 ComputeAndCompareR0<float>(&builder, 21.0f, {}, ErrorSpec(0.0001, 1e-4)); in XLA_TEST_F()
[all …]
Dselect_test.cc43 ComputeAndCompareR0<float>(&builder, 123.0f, {}, error_spec_); in TEST_F()
53 ComputeAndCompareR0<int32>(&builder, -42, {}); in TEST_F()
63 ComputeAndCompareR0<float>(&builder, 42.0f, {}, error_spec_); in TEST_F()
Dcall_test.cc83 ComputeAndCompareR0<float>(&builder, 42.0, {}, ErrorSpec(0.01f)); in XLA_TEST_F()
136 ComputeAndCompareR0<float>(&builder3, 10.0f, {start.get()}, ErrorSpec(0.0f)); in XLA_TEST_F()
Ddeep_graph_test.cc35 ComputeAndCompareR0<int32>(&b, /*expected=*/kDepth + 3, in TEST_F()
Dmap_test.cc179 ComputeAndCompareR0<float>(&builder, 43.0, {param0_data.get()}, in TEST_F()
339 ComputeAndCompareR0<float>(&builder, 73.0, {}, ErrorSpec(0.01f)); in XLA_TEST_F()
527 ComputeAndCompareR0<float>(&builder, 32.0f, in TEST_F()
554 ComputeAndCompareR0<float>( in TEST_F()
575 ComputeAndCompareR0<float>(&builder, 100.0f, {param0_data.get()}, in TEST_F()
Dcholesky_test.cc257 ComputeAndCompareR0<float>(&builder, 0.0, {input_data.get()}, in XLA_TEST_P()
305 ComputeAndCompareR0<float>(&builder, 0.0, in XLA_TEST_P()
Dwhile_test.cc78 ComputeAndCompareR0<int32>(&builder, 5, {}); in XLA_TEST_F()
114 ComputeAndCompareR0<int64>(&builder, 5, {}); in XLA_TEST_F()
147 ComputeAndCompareR0<int32>(&builder, 5, {}); in XLA_TEST_F()
177 ComputeAndCompareR0<bool>(&builder, true, {}); in XLA_TEST_F()
1029 ComputeAndCompareR0<float>(&outer, 43.0f, {parameter_data.get()}, in XLA_TEST_F()
1140 ComputeAndCompareR0<int32>(&builder, 42, {}); in XLA_TEST_F()
1186 ComputeAndCompareR0<int32>(&builder, 5, {}); in XLA_TEST_F()
1259 ComputeAndCompareR0<int32>(&builder, 2, {}); in XLA_TEST_F()
Dfloor_ceil_test.cc64 ComputeAndCompareR0<float>(&builder, expected, /*arguments=*/{}); in TestR0F32()
Dbitcast_convert_test.cc137 ComputeAndCompareR0<float>(&builder, 42.0f, {}); in TEST_F()
Dclient_library_test_base.h129 void ComputeAndCompareR0(XlaBuilder* builder, NativeT expected,
132 void ComputeAndCompareR0(XlaBuilder* builder, NativeT expected,
449 void ClientLibraryTestBase::ComputeAndCompareR0( in ComputeAndCompareR0() function
458 void ClientLibraryTestBase::ComputeAndCompareR0( in ComputeAndCompareR0() function
Dconstants_test.cc204 ComputeAndCompareR0<float>(&b, -1, {}, error_spec_); in TEST_F()
Dunary_op_test.cc160 ComputeAndCompareR0<float>(&builder, 8.5f, {}); in XLA_TEST_F()
Dparams_test.cc51 ComputeAndCompareR0<float>(&builder, 3.14159f, {param0_data.get()}, in XLA_TEST_F()
281 ComputeAndCompareR0<float>(&builder, target, param_data, ErrorSpec(0.0001f)); in XLA_TEST_F()
Ddot_operation_test.cc104 this->template ComputeAndCompareR0<T>(&builder, static_cast<T>(0.0), {}, in XLA_TYPED_TEST()
130 this->template ComputeAndCompareR0<T>(&builder, static_cast<T>(6.0f), {}, in XLA_TYPED_TEST()
141 this->template ComputeAndCompareR0<T>(&builder, static_cast<T>(29.5f), {}, in XLA_TYPED_TEST()
Dconvert_test.cc427 ComputeAndCompareR0<float>(&builder, 42.0f, {}, ErrorSpec(0.0001)); in TEST_F()

12