| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| constants_test.cc | 34 ComputeAndCompareR0<int32>(&builder, 4, {});
48 ComputeAndCompareR0<float>(&builder, -7, {});
50 ComputeAndCompareR0<float>(&builder, 0.5, {});
56 ComputeAndCompareR0<int32>(&builder, -3, {});
62 ComputeAndCompareR0<float>(&builder, -3.2, {});
68 ComputeAndCompareR0<int32>(&builder, 0, {});
74 ComputeAndCompareR0<float>(&builder, 0.0, {});
92 ComputeAndCompareR0<int32>(&builder, 1, {});
98 ComputeAndCompareR0<float>(&builder, 1., {});
104 ComputeAndCompareR0<float>(&builder, std::numeric_limits<float>::epsilon()
[all...] |
| self_adjoint_eig_test.cc | 236 ComputeAndCompareR0<float>(&builder, expected_vals, {a_data.get()},
259 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()},
272 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()},
285 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()},
298 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()},
311 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()},
|
| svd_test.cc | 214 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()},
226 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()},
238 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()},
250 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()},
262 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()},
274 ComputeAndCompareR0<float>(&builder, 1e-3, {a_data.get()},
|
| math_test.cc | 224 ComputeAndCompareR0<float>(&builder, 0.0f, {zero_data.get()}, error_spec_);
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| pred_test.cc | 40 ComputeAndCompareR0<bool>(&builder, expected, {});
47 ComputeAndCompareR0<bool>(&builder, true, {});
53 ComputeAndCompareR0<bool>(&builder, false, {});
100 ComputeAndCompareR0<bool>(&builder, true, {});
107 ComputeAndCompareR0<bool>(&builder, false, {});
114 ComputeAndCompareR0<bool>(&builder, false, {});
125 ComputeAndCompareR0<bool>(&builder, true, {});
136 ComputeAndCompareR0<bool>(&builder, false, {});
|
| scalar_computations_test.cc | 55 ComputeAndCompareR0<bool>(&builder, expected, {});
66 ComputeAndCompareR0<NativeT>(&builder, expected, {});
74 ComputeAndCompareR0<float>(&builder, 2.1f, {}, error_spec_);
81 ComputeAndCompareR0<float>(&builder, -2.1f, {}, error_spec_);
88 ComputeAndCompareR0<int32>(&builder, -2, {});
95 ComputeAndCompareR0<float>(&builder, 7.6f, {}, error_spec_);
102 ComputeAndCompareR0<int32>(&builder, 7, {});
109 ComputeAndCompareR0<uint32>(&builder, 92, {});
116 ComputeAndCompareR0<uint8>(&builder, 92, {});
125 ComputeAndCompareR0<uint64>(&builder, a + b, {})
[all...] |
| deep_graph_test.cc | 35 ComputeAndCompareR0<int32>(&b, /*expected=*/kDepth + 3,
|
| vector_ops_reduce_test.cc | 64 ComputeAndCompareR0<float>(&builder_, -4.2f, {}, errspec_);
78 ComputeAndCompareR0<float>(&builder_, expected, {}, errspec_);
89 ComputeAndCompareR0<float>(&builder_, 2.6f, {}, errspec_);
100 ComputeAndCompareR0<float>(&builder_, 4.0f, {}, errspec_);
200 ComputeAndCompareR0<float>(&builder_, 63.0, {}, errspec_);
|
| bfloat16_test.cc | 58 ComputeAndCompareR0<bfloat16>(&builder, static_cast<bfloat16>(3.0f), {},
67 ComputeAndCompareR0<bfloat16>(&builder, static_cast<bfloat16>(1.387f), {},
75 ComputeAndCompareR0<bfloat16>(&builder, static_cast<bfloat16>(-2.1f), {},
|
| conditional_test.cc | 188 ComputeAndCompareR0<float>(&builder, 56.0f, {pred_arg.get()}, error_spec_);
216 ComputeAndCompareR0<float>(&builder, expected, {branch_index_arg.get()},
231 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_);
268 ComputeAndCompareR0<float>(&builder, expected, {branch_index_arg.get()},
284 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_);
297 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_);
311 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_);
324 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_);
338 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_);
360 ComputeAndCompareR0<float>(&builder, 12.0f, {pred_arg.get()}, error_spec_)
[all...] |
| call_test.cc | 83 ComputeAndCompareR0<float>(&builder, 42.0, {}, ErrorSpec(0.01f));
136 ComputeAndCompareR0<float>(&builder3, 10.0f, {start.get()}, ErrorSpec(0.0f));
|
| floor_ceil_test.cc | 64 ComputeAndCompareR0<float>(&builder, expected, /*arguments=*/{});
|
| reduce_test.cc | 115 ComputeAndCompareR0<float>(&builder, expected, {input_global_data.get()},
150 ComputeAndCompareR0<bool>(&builder, expected, {input_global_data.get()});
224 ComputeAndCompareR0<float>(&builder, expected, {input_global_data.get()},
569 ComputeAndCompareR0<float>(&builder, expected, {}, ErrorSpec(0.0001));
581 ComputeAndCompareR0<float>(&builder, expected, {}, ErrorSpec(0.0001));
596 ComputeAndCompareR0<float>(&builder, input_max, {}, ErrorSpec(0.0001));
612 ComputeAndCompareR0<float>(&builder, input_min, {}, ErrorSpec(0.0001));
624 ComputeAndCompareR0<uint32>(&builder, 1, {});
636 ComputeAndCompareR0<uint32>(&builder, 2, {});
657 ComputeAndCompareR0<float>(&builder, 21.0f, {}, ErrorSpec(0.0001, 1e-4))
[all...] |
| select_test.cc | 43 ComputeAndCompareR0<float>(&builder, 123.0f, {}, error_spec_);
53 ComputeAndCompareR0<int32>(&builder, -42, {});
63 ComputeAndCompareR0<float>(&builder, 42.0f, {}, error_spec_);
|
| bitcast_convert_test.cc | 137 ComputeAndCompareR0<float>(&builder, 42.0f, {});
|
| cholesky_test.cc | 192 ComputeAndCompareR0<float>(&builder, 0.0, {input_data.get()},
|
| map_test.cc | 179 ComputeAndCompareR0<float>(&builder, 43.0, {param0_data.get()},
339 ComputeAndCompareR0<float>(&builder, 73.0, {}, ErrorSpec(0.01f));
527 ComputeAndCompareR0<float>(&builder, 32.0f,
554 ComputeAndCompareR0<float>(
575 ComputeAndCompareR0<float>(&builder, 100.0f, {param0_data.get()},
|
| while_test.cc | 78 ComputeAndCompareR0<int32>(&builder, 5, {});
114 ComputeAndCompareR0<int64>(&builder, 5, {});
147 ComputeAndCompareR0<int32>(&builder, 5, {});
177 ComputeAndCompareR0<bool>(&builder, true, {});
[all...] |
| client_library_test_base.h | 127 void ComputeAndCompareR0(XlaBuilder* builder, NativeT expected,
130 void ComputeAndCompareR0(XlaBuilder* builder, NativeT expected,
418 void ClientLibraryTestBase::ComputeAndCompareR0(
427 void ClientLibraryTestBase::ComputeAndCompareR0(
|
| constants_test.cc | 204 ComputeAndCompareR0<float>(&b, -1, {}, error_spec_);
|
| unary_op_test.cc | 160 ComputeAndCompareR0<float>(&builder, 8.5f, {});
|
| params_test.cc | 51 ComputeAndCompareR0<float>(&builder, 3.14159f, {param0_data.get()},
281 ComputeAndCompareR0<float>(&builder, target, param_data, ErrorSpec(0.0001f));
|
| dot_operation_test.cc | 97 this->template ComputeAndCompareR0<T>(&builder, static_cast<T>(0.0), {},
123 this->template ComputeAndCompareR0<T>(&builder, static_cast<T>(6.0f), {},
134 this->template ComputeAndCompareR0<T>(&builder, static_cast<T>(29.5f), {},
[all...] |
| convert_test.cc | 427 ComputeAndCompareR0<float>(&builder, 42.0f, {}, ErrorSpec(0.0001));
|
| array_elementwise_ops_test.cc | 146 ComputeAndCompareR0<bool>(&builder, false, {});
150 ComputeAndCompareR0<bool>(&builder, false, {});
154 ComputeAndCompareR0<bool>(&builder, false, {});
157 ComputeAndCompareR0<bool>(&builder, false, {});
160 ComputeAndCompareR0<bool>(&builder, true, {});
[all...] |
