Home
last modified time | relevance | path

Searched refs:ConstantR1 (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/tests/
Dselect_test.cc68 auto pred = ConstantR1<bool>(&builder, {}); in XLA_TEST_F()
69 auto on_true = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
70 auto on_false = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
78 auto pred = ConstantR1<bool>(&builder, {false, true, false, true, false}); in TEST_F()
80 ConstantR1<float>(&builder, {-2.5f, 25.5f, 2.25f, -10.0f, 6.0f}); in TEST_F()
82 ConstantR1<float>(&builder, {10.0f, 5.0f, 1.0f, 10.0f, -6.0f}); in TEST_F()
93 auto v1 = ConstantR1<int32>(&builder, {}); in XLA_TEST_F()
94 auto v2 = ConstantR1<int32>(&builder, {}); in XLA_TEST_F()
96 auto on_true = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
97 auto on_false = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
[all …]
Darray_elementwise_ops_test.cc55 auto a = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
63 auto a = ConstantR1<float>(&builder, {-2.5f, 3.14f, 2.25f, -10.0f, 6.0f}); in XLA_TEST_F()
72 auto a = ConstantR1<double>(&builder, {-2.5, 3.14, 2.25, -10.0, 6.0}); in XLA_TEST_F()
80 auto a = ConstantR1<int32>(&builder, in XLA_TEST_F()
96 auto a = ConstantR1<complex64>(&builder, {}); in XLA_TEST_F()
104 auto a = ConstantR1<complex64>( in XLA_TEST_F()
116 ConstantR1<int64>(&builder, { in XLA_TEST_F()
143 auto a = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
176 auto a = ConstantR1<float>(&builder, in XLA_TEST_F()
186 auto a = ConstantR1<float>(&builder, {-2.5f, 3.14f, 2.25f, -10.0f, 6.0f}); in XLA_TEST_F()
[all …]
Dvector_ops_simple_test.cc53 auto x = ConstantR1<float>( in XLA_TEST_F()
72 auto x = ConstantR1<float>(&builder, exponents); in XLA_TEST_F()
110 auto x = ConstantR1<float>( in XLA_TEST_F()
121 auto x = ConstantR1<int32>(&builder, {2, -2, 12, -4, 5, 20, -15, 0, -2, 1}); in XLA_TEST_F()
130 auto x = ConstantR1<uint32>( in XLA_TEST_F()
140 auto x = ConstantR1<float>(&builder, in XLA_TEST_F()
154 auto x = ConstantR1<float>( in XLA_TEST_F()
156 auto y = ConstantR1<float>( in XLA_TEST_F()
167 auto x = ConstantR1<float>( in XLA_TEST_F()
169 auto y = ConstantR1<float>( in XLA_TEST_F()
[all …]
Dtuple_test.cc88 ConstantR1<float>(&builder, constant_vector), in XLA_TEST_F()
103 {ConstantR0<float>(&builder, 7.0), ConstantR1<float>(&builder, {})}); in XLA_TEST_F()
127 Tuple(&builder, {ConstantR1<float>(&builder, constant_vector), in XLA_TEST_F()
139 {ConstantR1<float>(&builder, {}), in XLA_TEST_F()
147 auto value = ConstantR1<float>(&builder, {4.5f}); in XLA_TEST_F()
166 Tuple(&builder, {ConstantR1<float>(&builder, constant_vector), in XLA_TEST_F()
195 Tuple(&builder, {ConstantR1<float>(&builder, constant_vector), in XLA_TEST_F()
254 Tuple(&builder, {ConstantR1<float>(&builder, constant_vector), in XLA_TEST_F()
284 auto tuple12 = Tuple(&builder, {ConstantR1<float>(&builder, vec1), in XLA_TEST_F()
285 ConstantR1<float>(&builder, vec2)}); in XLA_TEST_F()
[all …]
Dconvert_test.cc50 auto a = ConstantR1<int32>(&builder, {42, 64}); in TEST_F()
59 auto a = ConstantR1<int32>(&builder, {42, 64}); in TEST_F()
68 auto a = ConstantR1<int32>(&builder, {42, 0, -64}); in TEST_F()
77 auto a = ConstantR1<uint32>(&builder, {42, 64}); in TEST_F()
86 auto a = ConstantR1<uint32>(&builder, {42, 64}); in TEST_F()
95 auto a = ConstantR1<uint32>(&builder, {42, 0, 64}); in TEST_F()
104 auto a = ConstantR1<float>(&builder, {42.0f, 64.0f}); in TEST_F()
113 auto a = ConstantR1<float>(&builder, {42.0f, 0.0f, 64.0f}); in TEST_F()
122 auto a = ConstantR1<int32>(&builder, {42, 64}); in TEST_F()
131 auto a = ConstantR1<bool>(&builder, {true, false, true}); in TEST_F()
[all …]
Dconcat_test.cc52 auto a = ConstantR1<float>(&builder, {42.0, 64.0}); in XLA_TEST_F()
61 auto a = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
83 auto a = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
84 auto b = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
93 auto a = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
94 auto b = ConstantR1<float>(&builder, {256.0}); in XLA_TEST_F()
103 auto a = ConstantR1<float>(&builder, {42.0, 64.0}); in XLA_TEST_F()
104 auto b = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
113 auto a = ConstantR1<float>(&builder, {42.0, 64.0}); in XLA_TEST_F()
114 auto b = ConstantR1<float>(&builder, {256.0}); in XLA_TEST_F()
[all …]
Dbitcast_convert_test.cc46 auto a = ConstantR1<int32>(&builder, {42, 64}); in TEST_F()
55 auto a = ConstantR1<float>(&builder, {42.0f, 64.0f}); in TEST_F()
65 ConstantR1<int32>(&builder, {0, static_cast<int32>(0x80000000), in TEST_F()
76 auto a = ConstantR1<int32>(&builder, {}); in XLA_TEST_F()
85 auto a = ConstantR1<float>(&builder, {42.6, 64.4}); in TEST_F()
94 auto a = ConstantR1<int32>(&builder, {std::numeric_limits<int32>::min(), in TEST_F()
107 auto a = ConstantR1<float>(&builder, {42.0f, 64.0f}); in TEST_F()
119 auto a = ConstantR1<int32>(&builder, {0x42280000, 0x42800000}); in TEST_F()
133 auto input = ConstantR1<int32>(&builder, {0x42280000}); in TEST_F()
Dunary_op_test.cc41 auto arg = ConstantR1<T>(&builder, {}); in AbsSize0TestHelper()
54 auto arg = ConstantR1<T>(&builder, {-2, 25, 0, -123, inf<T>(), -inf<T>()}); in AbsTestHelper()
63 auto arg = ConstantR1<T>( in SignTestHelper()
75 auto arg = ConstantR1<T>(&builder, {-2, 25, 0, -123}); in SignAbsTestHelper()
97 auto arg = ConstantR1<complex64>(&builder, {{-2, 0}, in AbsTestHelper()
113 auto arg = ConstantR1<complex64>( in SignTestHelper()
127 ConstantR1<complex64>(&builder, {{-2, 0}, {0, 25}, {0, 0}, {-0.4, 0.3}}); in SignAbsTestHelper()
205 auto lhs = ConstantR1<int32>(&builder, {0, 1}); in XLA_TEST_F()
206 auto rhs = ConstantR1<int32>(&builder, {1, 1}); in XLA_TEST_F()
214 auto lhs = ConstantR1<int32>(&builder, {0, 1}); in XLA_TEST_F()
[all …]
Ddeconstruct_tuple_test.cc57 auto const1 = ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in TEST_F()
58 auto const2 = ConstantR1<float>(&builder, {2.0, 4.0, 6.0, 8.0}); in TEST_F()
76 auto const1 = ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in TEST_F()
77 auto const2 = ConstantR1<float>(&builder, {2.0, 4.0, 6.0, 8.0}); in TEST_F()
106 auto const1 = ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in XLA_TEST_F()
107 auto const2 = ConstantR1<float>(&builder, {2.0, 4.0, 6.0, 8.0}); in XLA_TEST_F()
132 auto const1 = ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in TEST_F()
133 auto const2 = ConstantR1<float>(&builder, {2.0, 4.0, 6.0, 8.0}); in TEST_F()
162 ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in TEST_F()
188 auto const1 = ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in XLA_TEST_F()
[all …]
Daxpy_simple_test.cc33 auto x = ConstantR1<float>( in TEST_F()
46 auto x = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
47 auto y = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
58 auto x = ConstantR1<float>( in TEST_F()
60 auto y = ConstantR1<float>( in TEST_F()
Dselect_and_scatter_test.cc217 const auto operand = ConstantR1<float>(&builder_, {}); in XLA_TEST_F()
218 const auto source = ConstantR1<float>(&builder_, {}); in XLA_TEST_F()
228 ConstantR1<float>(&builder_, {1.f, 9.f, 3.f, 7.f, 5.f, 6.f}); in XLA_TEST_F()
229 const auto source = ConstantR1<float>(&builder_, {34.f, 42.f}); in XLA_TEST_F()
239 const auto operand = ConstantR1<int32>(&builder_, {-1, 0, 6, 4, -4, 10}); in XLA_TEST_F()
240 const auto source = ConstantR1<int32>(&builder_, {-10, 20}); in XLA_TEST_F()
250 const auto operand = ConstantR1<int32>(&builder_, {1, 9, 3, 7, 5, 6}); in XLA_TEST_F()
251 const auto source = ConstantR1<int32>(&builder_, {34, 42, 53, 19}); in XLA_TEST_F()
450 const auto operand = ConstantR1<float>(&builder_, {1, 2, 3, 100, 3, 2, 1}); in XLA_TEST_F()
451 const auto source = ConstantR1<float>(&builder_, {34, 42, 53, 19}); in XLA_TEST_F()
[all …]
Dbfloat16_test.cc92 auto scale = ConstantR1<bfloat16>( in XLA_TEST_F()
95 auto offset = ConstantR1<bfloat16>( in XLA_TEST_F()
124 auto scale = ConstantR1<bfloat16>( in XLA_TEST_F()
127 auto mean = ConstantR1<bfloat16>( in XLA_TEST_F()
130 auto var = ConstantR1<bfloat16>( in XLA_TEST_F()
Ddeallocation_test.cc69 ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0}); in TEST_F()
82 ConstantR1<float>(&builder, {}); in TEST_F()
96 ConstantR1<float>(&builder, {1.0, 2.0, 3.0})}); in XLA_TEST_F()
127 ConstantR1<float>(&builder, {1.0, 2.0, 3.0})}); in XLA_TEST_F()
128 Tuple(&builder, {inner_tuple, ConstantR1<float>(&builder, {0.123, 0.456})}); in XLA_TEST_F()
Dlocal_client_allocation_test.cc41 auto x = ConstantR1<float>(&builder, {0.0f, 1.0f, 2.0f}); in XLA_TEST_F()
42 auto y = ConstantR1<float>(&builder, {2.0f, 3.0f, 4.0f}); in XLA_TEST_F()
77 auto x = ConstantR1<float>(&builder, {0.0f, 1.0f, 2.0f}); in XLA_TEST_F()
78 auto y = ConstantR1<float>(&builder, {2.0f, 3.0f, 4.0f}); in XLA_TEST_F()
Dconstants_test.cc46 ConstantR1<float>(&builder, {}); in TEST_F()
55 ConstantR1<float>(&builder, constant); in TEST_F()
64 ConstantR1<int32>(&builder, constant); in TEST_F()
73 ConstantR1<uint32>(&builder, constant); in TEST_F()
82 ConstantR1<float>(&builder, constant); in TEST_F()
92 ConstantR1<float>(&builder, constant); in TEST_F()
Dwhile_test.cc143 Reduce(ConstantR1<int32>(&builder, 2, 1), ConstantR0<int32>(&builder, 0), in XLA_TEST_F()
218 auto input = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
225 auto init = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
272 auto input = ConstantR1<float>(&builder, 8, 0.125f); in XLA_TEST_F()
279 auto init = ConstantR1<float>(&builder, 8, 0.f); in XLA_TEST_F()
332 auto input = ConstantR1<float>(&builder, 8, 0.125f); in XLA_TEST_F()
339 auto init = ConstantR1<float>(&builder, 8, 0.f); in XLA_TEST_F()
392 ConstantR1<float>(&builder, 3, 1.f), in XLA_TEST_F()
393 ConstantR1<float>(&builder, 3, 2.f), in XLA_TEST_F()
394 ConstantR1<float>(&builder, 3, 3.f)}); in XLA_TEST_F()
[all …]
Dpred_test.cc82 ConstantR1<bool>(&builder, {true, false, false, true}); in TEST_F()
98 auto a = ConstantR1<bool>(&builder, {true, false}); in TEST_F()
105 auto a = ConstantR1<bool>(&builder, {false, false}); in TEST_F()
112 auto a = ConstantR1<bool>(&builder, {}); in TEST_F()
Dbatch_normalization_test.cc107 auto y = ConstantR1<float>(&builder, {3.14, 4.25}); in XLA_TEST_P()
151 auto set_means = ConstantR1<float>(&builder, {2.f, 4.2f}); in XLA_TEST_P()
164 auto variance = ConstantR1<float>(&builder, {6.f, .02f}); in XLA_TEST_P()
178 auto gamma = ConstantR1<float>(&builder, {1.0, 1.0}); in XLA_TEST_P()
179 auto beta = ConstantR1<float>(&builder, {0.0, 0.0}); in XLA_TEST_P()
197 auto epsilon2 = ConstantR1<float>(&builder, {kEpsilon, kEpsilon}); in XLA_TEST_P()
239 auto scale = ConstantR1<float>(&builder, {2.0f, 3.0f}); in XLA_TEST_P()
241 auto offset = ConstantR1<float>(&builder, {1.0f, 2.0f}); in XLA_TEST_P()
262 auto scale = ConstantR1<float>(&builder, {2.0f, 3.0f}); in XLA_TEST_P()
264 auto offset = ConstantR1<float>(&builder, {1.0f, 2.0f}); in XLA_TEST_P()
[all …]
Dbroadcast_simple_test.cc150 Broadcast(ConstantR1<float>(&b, {1, 2, 3}), {2});
164 BroadcastInDim(ConstantR1<float>(&b, {1, 2}), {2, 2}, {1});
177 BroadcastInDim(ConstantR1<float>(&b, {1, 2}), {2, 2}, {0});
226 BroadcastInDim(ConstantR1<float>(&b, {1, 2}), {3, 2}, {1});
268 Broadcast(ConstantR1<float>(&b, {}), {2}); in XLA_TEST_F()
276 Broadcast(ConstantR1<float>(&b, {1, 2, 3}), {0});
651 auto r1 = ConstantR1<float>(&b, {10, 20}); in XLA_TEST_F()
664 auto r1 = ConstantR1<float>(&b, {10, 20}); in XLA_TEST_F()
677 auto r1 = ConstantR1<float>(&b, {10, 20}); in XLA_TEST_F()
690 auto r1_0 = ConstantR1<float>(&b, {1000, 2000}); in XLA_TEST_F()
[all …]
Dvector_ops_reduce_test.cc59 auto x = ConstantR1<float>( in TEST_F()
73 auto x = ConstantR1<float>(&builder_, input); in TEST_F()
84 auto x = ConstantR1<float>( in TEST_F()
95 auto x = ConstantR1<float>( in TEST_F()
Dconditional_test.cc399 auto operand1 = ConstantR1<float>(&builder, {24.0f, 56.0f}); in XLA_TEST_F()
400 auto operand2 = ConstantR1<float>(&builder, {10.0f, 11.0f}); in XLA_TEST_F()
418 auto operand1 = ConstantR1<float>(&builder, {24.0f, 56.0f}); in XLA_TEST_P()
419 auto operand2 = ConstantR1<float>(&builder, {10.0f, 11.0f}); in XLA_TEST_P()
455 auto operand1 = ConstantR1<float>(&builder, {24.0f, 56.0f}); in XLA_TEST_F()
456 auto operand2 = ConstantR1<float>(&builder, {10.0f, 11.0f}); in XLA_TEST_F()
488 Tuple(&builder, {ConstantR1<float>(&builder, {12.2f, 15.8f}), in XLA_TEST_F()
489 ConstantR1<float>(&builder, {25.6f, 29.2f})}); in XLA_TEST_F()
508 auto true_array = ConstantR1<float>(&true_builder, {12.8f, 14.6f}); in XLA_TEST_F()
519 auto false_array = ConstantR1<float>(&false_builder, {26.4f, 32.6f}); in XLA_TEST_F()
[all …]
Dlocal_client_execute_test.cc82 auto y = ConstantR1<float>(&builder, {}); in XLA_TEST_F()
95 auto y = ConstantR1<float>(&builder, {2.0f, 3.0f, 4.0f}); in XLA_TEST_F()
109 auto y = ConstantR1<float>(&builder, {2.0f, 3.0f, 4.0f}); in XLA_TEST_F()
740 auto tuple12 = Tuple(&builder, {ConstantR1<float>(&builder, vec1), in XLA_TEST_F()
741 ConstantR1<float>(&builder, vec2)}); in XLA_TEST_F()
742 auto tuple21 = Tuple(&builder, {ConstantR1<float>(&builder, vec2), in XLA_TEST_F()
743 ConstantR1<float>(&builder, vec1)}); in XLA_TEST_F()
758 auto y = ConstantR1<float>(&builder, {2.0f, 3.0f, 4.0f}); in XLA_TEST_F()
791 auto y = ConstantR1<float>(&builder, {2.0f, 3.0f, 4.0f}); in XLA_TEST_F()
792 auto z = ConstantR1<float>(&builder, {5.0f, 6.0f, 7.0f}); in XLA_TEST_F()
[all …]
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dmath_test.cc69 ConstantR1<T>(&b, { in TestIsInfOrNan()
242 auto x = ConstantR1<double>( in XLA_TEST_F()
263 auto x = ConstantR1<float>( in XLA_TEST_F()
274 auto x = ConstantR1<float>( in XLA_TEST_F()
286 auto x = ConstantR1<float>(&builder, {0.0, -0.0}); in XLA_TEST_F()
294 auto x = ConstantR1<float>(&builder, {16.0, 1.0, 1024.0, 0.16, 0.2, 12345}); in XLA_TEST_F()
303 auto x = ConstantR1<float>(&builder, {1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 0.5, 1.5, in XLA_TEST_F()
332 auto x = ConstantR1<half>(&b, { in XLA_TEST_F()
349 auto x = ConstantR1<float>(&builder, {1.0, 0.5, 1 / 3.0, 0.25, 1 / 6.0, 0.125, in XLA_TEST_F()
404 ConstantR1<float>(&builder, {nan, nan, 0.53327996, -6.00773744602e+37, in XLA_TEST_F()
[all …]
Dsorting_test.cc34 ConstantR1<float>(&builder, {0.0, 1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0}); in XLA_TEST_F()
42 ConstantR1<float>(&builder, {7.0, 6.0, 5.0, 4.0, 3.0, 2.0, 1.0, 0.0}); in XLA_TEST_F()
49 auto x_rev = ConstantR1<int>(&builder, {std::numeric_limits<int>::min(), in XLA_TEST_F()
64 auto x = ConstantR1<float>(&builder, inputs); in XLA_TEST_F()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dresampler_ops.cc203 xla::ConstantR1<float>(ctx->builder(), {-1, -1}), warp_type), in BoundSamples()
208 xla::ConstantR1<float>( in BoundSamples()
312 auto shifting_value = xla::ConstantR1<int32>( in CalculateGradData()
385 auto shifting_value = xla::ConstantR1<int32>( in CalculateGradWarp()
413 xla::ConstantR1<float>(ctx->builder(), {0, 0, -1, 1}), data_type), in CalculateGradWarp()
421 xla::ConstantR1<float>(ctx->builder(), {-1, 1, 0, 0}), data_type), in CalculateGradWarp()
429 xla::ConstantR1<float>(ctx->builder(), {0, -1, 0, 1}), data_type), in CalculateGradWarp()
437 xla::ConstantR1<float>(ctx->builder(), {-1, 0, 1, 0}), data_type), in CalculateGradWarp()
560 xla::ConstantR1<float>( in Compile()

12